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 branches found
No related tags found
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