Skip to content
Snippets Groups Projects
Commit fd6fdd75 authored by Tim O'Donnell's avatar Tim O'Donnell
Browse files

remove spurious absolute mhcflurry import

parent 8f08993d
No related merge requests found
......@@ -7,7 +7,7 @@ import sklearn
import numpy
import scipy
import mhcflurry
from ..regression_target import ic50_to_regression_target
def make_scores(
......@@ -38,8 +38,7 @@ def make_scores(
dict with entries "auc", "f1", "tau"
"""
y_pred = mhcflurry.regression_target.ic50_to_regression_target(
ic50_y_pred, max_ic50)
y_pred = ic50_to_regression_target(ic50_y_pred, max_ic50)
try:
auc = sklearn.metrics.roc_auc_score(
ic50_y <= threshold_nm,
......
......@@ -26,8 +26,6 @@ import math
import numpy
import pandas
import mhcflurry
from .scoring import make_scores
from .class1_binding_predictor import Class1BindingPredictor
from ..hyperparameters import HyperparameterDefaults
......@@ -186,7 +184,7 @@ def train_and_test_one_model_one_fold(
impute,
model_description))
predictor = mhcflurry.Class1BindingPredictor(
predictor = Class1BindingPredictor(
max_ic50=max_ic50,
**model_params)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment