summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xtraining/dtrain/parallelize.rb3
1 files changed, 2 insertions, 1 deletions
diff --git a/training/dtrain/parallelize.rb b/training/dtrain/parallelize.rb
index acfd7290..30fb0008 100755
--- a/training/dtrain/parallelize.rb
+++ b/training/dtrain/parallelize.rb
@@ -21,6 +21,7 @@ opts = Trollop::options do
opt :references, "references", :type => :string
opt :qsub, "use qsub", :type => :bool, :default => false
opt :dtrain_binary, "path to dtrain binary", :type => :string
+ opt :lplp_args, "arguments for lplp arguments", :type => :string, :default => "l2 select_k 100000"
end
puts opts.to_s
@@ -34,7 +35,7 @@ else
end
ruby = '/usr/bin/ruby'
lplp_rb = "#{dtrain_dir}/hstreaming/lplp.rb"
-lplp_args = 'l2 select_k 100000'
+lplp_args = opts[:lplp_args]
cat = '/bin/cat'
ini = opts[:config]