Skip to content

Commit

Permalink
general lint
Browse files Browse the repository at this point in the history
  • Loading branch information
vpchung committed Jul 3, 2024
1 parent 66a4f35 commit eda41f7
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions score.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,14 +5,14 @@
- ROC curve
- PR curve
"""
from glob import glob
import argparse
import json
import os
from glob import glob

import pandas as pd
import numpy as np
from sklearn.metrics import roc_auc_score, average_precision_score
import pandas as pd
from sklearn.metrics import average_precision_score, roc_auc_score

GOLDSTANDARD_COLS = {"epr_number": str, "disease_probability": str}
PREDICTION_COLS = {"epr_number": str, "disease_probability": np.float64}
Expand Down Expand Up @@ -53,7 +53,6 @@ def extract_gs_file(folder):
"Expected exactly one gold standard file in folder. "
f"Got {len(files)}. Exiting."
)

return files[0]


Expand Down

0 comments on commit eda41f7

Please sign in to comment.