sohomghosh
/

finrad_model

Model card Files Files and versions

sohomghosh commited on Jun 27, 2022

Commit

17ef62c

·

1 Parent(s): 989fadb

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -100,7 +100,7 @@ def do_predict(tokenizer):
   test_set = Triage(test_df, tokenizer, MAX_LEN, text_col_name)
   test_params = {'batch_size' : BATCH_SIZE, 'shuffle': False, 'num_workers':0}
   test_loader = DataLoader(test_set, **test_params)
-  out_stack, mask_stack = scoring_data_prep(dataset = validation_set)
   n = 0
   combined_output = []
   model.eval()

   test_set = Triage(test_df, tokenizer, MAX_LEN, text_col_name)
   test_params = {'batch_size' : BATCH_SIZE, 'shuffle': False, 'num_workers':0}
   test_loader = DataLoader(test_set, **test_params)
+  out_stack, mask_stack = scoring_data_prep(dataset = test_set)
   n = 0
   combined_output = []
   model.eval()