@@ -342,16 +342,16 @@ def train(
342
342
@cli .command ("test" )
343
343
@click .argument ("test_path" , type = click .Path (dir_okay = False , file_okay = True , exists = True ))
344
344
@click .argument ("models" , nargs = - 1 , type = click .Path (dir_okay = False , file_okay = True , exists = True ))
345
- @click .option ("--csv_file" , default = None , type = click .File (mode = "w" ), help = "CSV target" )
345
+ # @click.option("--csv_file", default=None, type=click.File(mode="w"), help="CSV target")
346
346
@click .option ("--batch_size" , type = int , default = 32 , help = "Size of batches" )
347
347
@click .option ("--device" , default = "cpu" , help = "Device to use for the network (cuda, cpu, etc.)" )
348
348
@click .option ("--debug" , default = False , is_flag = True )
349
- @click .option ("--verbose" , default = False , is_flag = True , help = "Print classification report" )
349
+ # @click.option("--verbose", default=False, is_flag=True, help="Print classification report")
350
350
@click .option ("--workers" , default = 1 , type = int , help = "Number of workers to use to load data" )
351
351
@click .option ("--avg" , default = "macro" , type = click .Choice (["micro" , "macro" ]), help = "Type of avering method to use on "
352
352
"metrics" )
353
- def test (test_path , models , csv_file , batch_size , device , debug , verbose , workers : int , avg : str ):
354
- """ Train one or more models according to [CONFIG_FILES] JSON configurations """
353
+ def test (test_path , models , batch_size , device , debug , workers : int , avg : str ):
354
+ """ Test one or many [MODELS] on the file at TEST_PATH """
355
355
if debug :
356
356
logger .setLevel (logging .DEBUG )
357
357
else :
0 commit comments