Commit
•
dc23c5f
1
Parent(s):
63275a8
Update trainer output dir (#427)
Browse files- Update trainer output dir (414cf94616fe848b0d773ec954db7fd83723e2e2)
Co-authored-by: Han Chen <[email protected]>
- geneformer/classifier.py +1 -1
geneformer/classifier.py
CHANGED
@@ -1142,7 +1142,7 @@ class Classifier:
|
|
1142 |
metric="eval_macro_f1",
|
1143 |
metric_columns=["loss", "eval_loss", "eval_accuracy", "eval_macro_f1"],
|
1144 |
),
|
1145 |
-
|
1146 |
)
|
1147 |
|
1148 |
return trainer
|
|
|
1142 |
metric="eval_macro_f1",
|
1143 |
metric_columns=["loss", "eval_loss", "eval_accuracy", "eval_macro_f1"],
|
1144 |
),
|
1145 |
+
storage_path=output_directory,
|
1146 |
)
|
1147 |
|
1148 |
return trainer
|