1 files changed, 6 insertions, 6 deletions

diff --git a/scripts/train_dic_creator.pl b/scripts/train_dic_creator.pl
index a228044..0a9f931 100755
--- a/scripts/train_dic_creator.pl
+++ b/scripts/train_dic_creator.pl
@@ -33,7 +33,7 @@ use strict;
 use warnings;
 use List::Util qw(shuffle);
 
-my ($source_lang, $target_lang, $cutoff, $dict_dir) = @ARGV;
+my ($source_lang, $target_lang, $dict_dir, $cutoff) = @ARGV;
 
 if (not defined $source_lang or not defined $target_lang) {
     die "usage: ./train_dic_creator.pl <source_lang> <target_lang> (cutoff)";
@@ -44,7 +44,7 @@ if (not defined $cutoff && $cutoff ne '') {
 }
 
 if (not defined $dict_dir && $dict_dir ne '') {
-    $dict_dir = './dictionaries/';
+    $dict_dir = '../dictionaries';
 }
 
 my $flipped = 0;
@@ -106,11 +106,11 @@ my $size = $cutoff / 2;
 my @head = @result[0..$size - 1];
 my @tail = @result[-$size..-1];
 
-my $train_file_name = $source_lang . '_' . $target_lang . '.train';
-my $test_file_name = $source_lang . '_' . $target_lang . '.test';
+my $train_file_name = "$dict_dir/$source_lang_$target_lang.train';
+my $test_file_name  = "$dict_dir/$source_lang_$target_lang.test';
 
-open my $train_fh, '>', $dict_dir . $train_file_name;
-open my $test_fh, '>', $dict_dir . $test_file_name;
+open my $train_fh, '>', $train_file_name;
+open my $test_fh, '>', $test_file_name;
 
 print $train_fh join("\n", @head);
 print $test_fh join("\n", @tail);