def mhc_binding_prediction(mutated_regions, alleles):
if args.random_mhc:
return mhc_random.generate_scored_epitopes(mutated_regions, alleles)
elif args.iedb_mhc:
mhc = IEDB_MHC1(alleles=alleles)
return mhc.predict(mutated_regions)
elif args.netmhc_cons:
predictor = ConsensusBindingPredictor(alleles)
return predictor.predict(mutated_regions)
else:
predictor = PanBindingPredictor(alleles)
return predictor.predict(mutated_regions)
def score_epitopes_mhc_cons(transcripts_df, alleles):
predictor = ConsensusBindingPredictor(alleles)
return predictor.predict(transcripts_df)
def make_mhc_predictor():
if args.netmhc_cons:
return ConsensusBindingPredictor(hla_allele_names)
else:
return PanBindingPredictor(hla_allele_names)