240808

arxiv submission
ssocean · Aug 7, 2024 · 3ac7184 · 3ac7184
1 parent eddd83f
commit 3ac7184
Show file tree

Hide file tree

Showing 132 changed files with 588 additions and 941,055 deletions.
diff --git a/.gitignore b/.gitignore
@@ -45,3 +45,5 @@ results/*
 official_runs/*
 *.0
 official_runs*
+*.pt
+*.pth
diff --git a/.idea/misc.xml b/.idea/misc.xml
diff --git a/.idea/modules.xml b/.idea/modules.xml
diff --git a/CACHE/CACHE_Config.py b/CACHE/CACHE_Config.py
@@ -1,8 +1,8 @@
 import os.path
 
-# base_dir = r'C:\Users\Ocean\Documents\GitHub\Dynamic_Literature_Review\CACHE'#'../CACHE/'
+# base_dir = r'~\Documents\GitHub\Dynamic_Literature_Review\CACHE'#'../CACHE/'
 base_dir = os.path.dirname(os.path.abspath(__file__))
-# base_dir = r'C:\Users\Ocean\Documents\GitHub\Dynamic_Literature_Review\test_cache'
+# base_dir = r'~\Documents\GitHub\Dynamic_Literature_Review\test_cache'
 def generate_cache_file_name(url='',force_file_name=None):
     if not force_file_name:
         if 'authors?' in url:

diff --git a/Openreview/Access.csv b/Openreview/Access.csv
diff --git a/Openreview/DKE.csv b/Openreview/DKE.csv
diff --git a/Openreview/IEEECommLTR.csv b/Openreview/IEEECommLTR.csv
diff --git a/Openreview/IET.csv b/Openreview/IET.csv
diff --git a/Openreview/TPAMI.csv b/Openreview/TPAMI.csv
diff --git a/Openreview/iclr_23_submissions.csv b/Openreview/iclr_23_submissions.csv
diff --git a/Openreview/nips_23_submissions.csv b/Openreview/nips_23_submissions.csv
diff --git a/README.md b/README.md
@@ -1 +1 @@
-# ScitePredict
+# ScImpactPredict
diff --git a/avg_journal_score_pred.py b/avg_journal_score_pred.py
@@ -27,14 +27,14 @@
 np.random.seed(seed)
 random.seed(seed)
 
-# 设置 cuDNN 确定性和禁用自动优化
+
 torch.backends.cudnn.deterministic = True
 torch.backends.cudnn.benchmark = False
 
 def save_args_to_json(args, file_path):
-    # 将 args 对象转换为字典
+
     args_dict = vars(args)
-    # 将字典保存到 JSON 文件
+
     with open(file_path, 'w') as f:
         json.dump(args_dict, f, indent=4)
 def get_args():
@@ -45,8 +45,8 @@ def get_args():
     parser.add_argument('--batch_size', type=int, default=20, help='Batch size for training and validation')
 
 
-    parser.add_argument('--data_path', type=str, default='/home/u1120220285/ScitePredict/data/Data_TNCSI_S_OA_AuthorCite_8242_fix1.csv', help='Path to the dataset CSV file')
-    parser.add_argument('--checkpoint', type=str, default='/home/u1120220285/llama3_weight', help='Model checkpoint path')
+    parser.add_argument('--data_path', type=str, default='ScImpactPredict/data/Data_TNCSI_S_OA_AuthorCite_8242_fix1.csv', help='Path to the dataset CSV file')
+    parser.add_argument('--checkpoint', type=str, default='llama3_weight', help='Model checkpoint path')
     parser.add_argument('--weight_dir', type=str, default='runs/Jul12_14-54-26_gpu22', help='Model checkpoint path')
     parser.add_argument('--loss_func', type=str, default='bce',choices=['bce','mse','l1'])
     parser.add_argument('--prompt_style', type=int,default=0)
@@ -66,11 +66,8 @@ def get_args():
     parser.add_argument('--lora_bias', type=str, default='none', help='Bias mode for LoRA layers')
     parser.add_argument('--target_modules', type=str, default='q_proj,v_proj', help='Comma-separated list of transformer modules to apply LoRA')
 
-
-
-    # 默认 TensorBoard 日志目录使用当前日期和时间
     default_tb_dir = datetime.now().strftime("%m-%d-%H-%M")
-    parser.add_argument('--runs_dir', type=str, default=os.path.join('/home/u1120220285/ScitePredict/inference',default_tb_dir), help='Directory for storing TensorBoard logs')
+    parser.add_argument('--runs_dir', type=str, default=os.path.join('ScImpactPredict/inference',default_tb_dir), help='Directory for storing TensorBoard logs')
 
     return parser.parse_args()
 
@@ -133,7 +130,7 @@ def __getitem__(self, idx):
 device_map={'':torch.cuda.current_device()}
 
 
-model = AutoPeftModelForSequenceClassification.from_pretrained(args.weight_dir, num_labels=args.num_labels, load_in_8bit=args.load_in_8bit,device_map=device_map,)# 不改变score 权重
+model = AutoPeftModelForSequenceClassification.from_pretrained(args.weight_dir, num_labels=args.num_labels, load_in_8bit=args.load_in_8bit,device_map=device_map,) 
 model.config.pad_token_id = model.config.eos_token_id
 model.loss_func = args.loss_func
 

diff --git a/data/Data_TNCSI_S.csv b/data/Data_TNCSI_S.csv
diff --git a/data/Data_TNCSI_S_OA_AuthorCite_8242.csv b/data/Data_TNCSI_S_OA_AuthorCite_8242.csv
diff --git a/data/Data_TNCSI_S_OA_AuthorCite_8242_fix0.csv b/data/Data_TNCSI_S_OA_AuthorCite_8242_fix0.csv
diff --git a/data/Data_TNCSI_S_OA_AuthorCite_8242_fix1.csv b/data/Data_TNCSI_S_OA_AuthorCite_8242_fix1.csv
diff --git a/data/NAID_2021_full.csv b/data/NAID_2021_full.csv
diff --git a/data/NAID_2022.csv b/data/NAID_2022.csv
diff --git a/data/NAID_2022_full.csv b/data/NAID_2022_full.csv
diff --git a/data/NAID_2022_test.csv b/data/NAID_2022_test.csv
diff --git a/data/NAID_2022_train.csv b/data/NAID_2022_train.csv
diff --git a/data/NAID_2023.csv b/data/NAID_2023.csv
diff --git a/data/NAID_2023_full.csv b/data/NAID_2023_full.csv
diff --git a/data/NAID_2023_test.csv b/data/NAID_2023_test.csv
diff --git a/data/NAID_2023_train.csv b/data/NAID_2023_train.csv
diff --git a/data/NAID_8000_2022.csv b/data/NAID_8000_2022.csv
diff --git a/data/NAID_8000_2022and2023.csv b/data/NAID_8000_2022and2023.csv
diff --git a/data/NAID_8000_2023.csv b/data/NAID_8000_2023.csv
diff --git a/data/NAID_for_build_test.csv b/data/NAID_for_build_test.csv
diff --git a/data/data_for_model.csv b/data/data_for_model.csv
diff --git a/data/data_for_model_balanced.csv b/data/data_for_model_balanced.csv
diff --git a/data/data_for_model_balanced_2.csv b/data/data_for_model_balanced_2.csv
-Original file line number
+Diff line change
@@ Expand Up / @@ -45,3 +45,5 @@ results/* @@
     official_runs/*
     *.0
     official_runs*
+    *.pt
+    *.pth