Exemple #1
0
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)
Exemple #2
0
def score_epitopes_mhc_cons(transcripts_df, alleles):
    predictor = ConsensusBindingPredictor(alleles)
    return predictor.predict(transcripts_df)
Exemple #3
0
 def make_mhc_predictor():
     if args.netmhc_cons:
         return ConsensusBindingPredictor(hla_allele_names)
     else:
         return PanBindingPredictor(hla_allele_names)