We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0b46ab8 commit 1d31e2fCopy full SHA for 1d31e2f
chebai/result/utils.py
@@ -116,7 +116,7 @@ def evaluate_model(
116
data_list = data_df.to_dict(orient="records")
117
else:
118
data_list = data_module.load_processed_data("test", filename)
119
- torch.save(data_list,"./data_list.pt")
+ torch.save(data_list,"./data_list"+ kind +".pt")
120
data_list = data_list[: data_module.data_limit]
121
preds_list = []
122
labels_list = []
0 commit comments