Commits

Elias Ponvert committed 2f554a7

trying to make separate test file type cli arg

Comments (0)

Files changed (1)

     op.add_option('-t', '--train')
     op.add_option('-T', '--input_type')
     op.add_option('-s', '--test')
+    op.add_option('-U', '--test_input_type')
     op.add_option('-o', '--output')
     op.add_option('-r', '--reverse', action='store_true')
     op.add_option('-f', '--filter_test', type='int', default='-1')
       log('guessing input type = ' + input_type)
       return input_type
 
+  def test_input_type(self):
+    if self.opt.test_input_type == None:
+      return self.input_type()
+    input_type_map = {'wsj':'WSJ', 'negra':'NEGRA', 'ctb':'CTB', 'brown':'WSJ'}
+    try:
+      return input_type_map[self.opt.test_input_type]
+    except KeyError:
+      input_type = guess_input_type(self._get_test_str())
+      log('guessing input type = ' + input_type)
+      return input_type
+
+
+
+
   def check_output(self):
     opt = self.opt
     if opt.output is not None:
     if self.opt.stats:
       return ' -stats '
     cmd = ' -test ' + self._get_test_str()
-    cmd += ' -testFileType ' + self.input_type()
+    cmd += ' -testFileType ' + self.test_input_type()
     return cmd
 
   def starter_train_out(self):