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 ...@@ -7,7 +7,7 @@ import sklearn
import numpy import numpy
import scipy import scipy
import mhcflurry from ..regression_target import ic50_to_regression_target
def make_scores( def make_scores(
...@@ -38,8 +38,7 @@ def make_scores( ...@@ -38,8 +38,7 @@ def make_scores(
dict with entries "auc", "f1", "tau" dict with entries "auc", "f1", "tau"
""" """
y_pred = mhcflurry.regression_target.ic50_to_regression_target( y_pred = ic50_to_regression_target(ic50_y_pred, max_ic50)
ic50_y_pred, max_ic50)
try: try:
auc = sklearn.metrics.roc_auc_score( auc = sklearn.metrics.roc_auc_score(
ic50_y <= threshold_nm, ic50_y <= threshold_nm,
......
...@@ -26,8 +26,6 @@ import math ...@@ -26,8 +26,6 @@ import math
import numpy import numpy
import pandas import pandas
import mhcflurry
from .scoring import make_scores from .scoring import make_scores
from .class1_binding_predictor import Class1BindingPredictor from .class1_binding_predictor import Class1BindingPredictor
from ..hyperparameters import HyperparameterDefaults from ..hyperparameters import HyperparameterDefaults
...@@ -186,7 +184,7 @@ def train_and_test_one_model_one_fold( ...@@ -186,7 +184,7 @@ def train_and_test_one_model_one_fold(
impute, impute,
model_description)) model_description))
predictor = mhcflurry.Class1BindingPredictor( predictor = Class1BindingPredictor(
max_ic50=max_ic50, max_ic50=max_ic50,
**model_params) **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