add pred result

truonghm · Sep 21, 2023 · 47b1c36 · 47b1c36
1 parent fe4b294
commit 47b1c36
Show file tree

Hide file tree

Showing 2 changed files with 2,058 additions and 0 deletions.
diff --git a/lib/codebert-bimodal/inference.sh b/lib/codebert-bimodal/inference.sh
@@ -0,0 +1,11 @@
+python lib/codebert-bimodal/run_classifier.py \
+			--model_type roberta \
+			--do_predict \
+			--test_file test_set_new.jsonl \
+			--max_seq_length 200 \
+			--per_gpu_eval_batch_size 2 \
+			--data_dir ./data/exp \
+			--output_dir ./models/checkpoint-best-aver/ \
+			--encoder_name_or_path microsoft/codebert-base \
+			--pred_model_dir ./models/checkpoint-last/ \
+			--prediction_file .logs/predictions_codebert_bimodal.txt