diff --git a/mhcflurry/class1_allele_specific/scoring.py b/mhcflurry/class1_allele_specific/scoring.py index e50ba83a4b616dcdbf55f971c31d5199093d329a..485d4303af62ea5a57e4e9b08e763640b74224a4 100644 --- a/mhcflurry/class1_allele_specific/scoring.py +++ b/mhcflurry/class1_allele_specific/scoring.py @@ -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, diff --git a/mhcflurry/class1_allele_specific/train.py b/mhcflurry/class1_allele_specific/train.py index 7e5824b88163e60d22a846c32adb6b0582297457..2c6d651924be67ecd43d2d638781e1b37461a9dd 100644 --- a/mhcflurry/class1_allele_specific/train.py +++ b/mhcflurry/class1_allele_specific/train.py @@ -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)