From 32dd546133a6d71e256fe2ecf6fcd1d607d461f1 Mon Sep 17 00:00:00 2001
From: zhongfa1999 <zhongfa1999@example.org>
Date: Mon, 21 Oct 2024 17:41:34 +0800
Subject: [PATCH] ADD file via upload

---
 LineVul-main/linevul/test.sh | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 LineVul-main/linevul/test.sh

diff --git a/LineVul-main/linevul/test.sh b/LineVul-main/linevul/test.sh
new file mode 100644
index 0000000..0b8c50a
--- /dev/null
+++ b/LineVul-main/linevul/test.sh
@@ -0,0 +1,18 @@
+python linevul_main_predict_test.py \
+  --output_dir=./saved_models \
+  --model_type=roberta \
+  --model_name=model.bin \
+  --tokenizer_name=microsoft/codebert-base \
+  --model_name_or_path=microsoft/codebert-base \
+  --do_test \
+  --train_data_file=../data/big-vul_dataset/new_train.csv \
+  --eval_data_file=../data/big-vul_dataset/new_valid.csv \
+  --test_data_file=../data/big-vul_dataset/test.csv \
+  --epochs 10 \
+  --block_size 512 \
+  --train_batch_size 16 \
+  --eval_batch_size 16 \
+  --learning_rate 2e-5 \
+  --max_grad_norm 1.0 \
+  --evaluate_during_training \
+  --seed 123456
\ No newline at end of file