Skip to content

Commit 1d31e2f

Browse files
committed
save all data_lists
1 parent 0b46ab8 commit 1d31e2f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

chebai/result/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def evaluate_model(
116116
data_list = data_df.to_dict(orient="records")
117117
else:
118118
data_list = data_module.load_processed_data("test", filename)
119-
torch.save(data_list,"./data_list.pt")
119+
torch.save(data_list,"./data_list"+ kind +".pt")
120120
data_list = data_list[: data_module.data_limit]
121121
preds_list = []
122122
labels_list = []

0 commit comments

Comments
 (0)