Update trainer output dir
Browse fileslocal_dir deprecated, change to storage_path
- 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
|