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 1581693 commit 96ef82eCopy full SHA for 96ef82e
chebai/result/utils.py
@@ -116,6 +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")
120
data_list = data_list[: data_module.data_limit]
121
preds_list = []
122
labels_list = []
@@ -159,7 +160,7 @@ def evaluate_model(
159
160
test_labels = _concat_tuple(labels_list)
161
return test_preds, test_labels
162
return test_preds, None
- elif len(preds_list) < 0:
163
+ elif len(preds_list) > 0:
164
if len(preds_list) > 0 and preds_list[0] is not None:
165
torch.save(
166
_concat_tuple(preds_list),
0 commit comments