Skip to content

Commit

Permalink
chore(LocusToGeneModel): remove evaluation outside experiment tracking
Browse files Browse the repository at this point in the history
  • Loading branch information
ireneisdoomed committed Dec 13, 2023
1 parent dc4bb53 commit bd67c78
Showing 1 changed file with 0 additions and 23 deletions.
23 changes: 0 additions & 23 deletions src/otg/method/l2g/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -223,30 +223,7 @@ def evaluate(
labelCol="label", predictionCol="prediction"
)

print("Evaluating model...") # noqa: T201
print( # noqa: T201
"... Area under ROC curve:",
binary_evaluator.evaluate(
results, {binary_evaluator.metricName: "areaUnderROC"}
),
)
print( # noqa: T201
"... Area under Precision-Recall curve:",
binary_evaluator.evaluate(
results, {binary_evaluator.metricName: "areaUnderPR"}
),
)
print( # noqa: T201
"... Accuracy:",
multi_evaluator.evaluate(results, {multi_evaluator.metricName: "accuracy"}),
)
print( # noqa: T201
"... F1 score:",
multi_evaluator.evaluate(results, {multi_evaluator.metricName: "f1"}),
)

if wandb_run_name and training_data:
print("Logging to W&B...") # noqa: T201
run = wandb.init(
project=self.wandb_l2g_project_name,
config=hyperparameters,
Expand Down

0 comments on commit bd67c78

Please sign in to comment.