diff --git a/mhcflurry/class1.py b/mhcflurry/class1.py new file mode 100644 index 0000000000000000000000000000000000000000..f90e79ec5852071f4f7264dea889633a89c4f9ef --- /dev/null +++ b/mhcflurry/class1.py @@ -0,0 +1,11 @@ +import pandas as pd + +from .mhc1_binding_predictor import Mhc1BindingPredictor + +def predict(alleles, peptides): + allele_dataframes = [] + for allele in alleles: + model = Mhc1BindingPredictor(allele=allele) + df = model.predict_peptides(peptides) + allele_dataframes.append(df) + return pd.concat(allele_dataframes)