cast_actor.py

from collections import namedtuple

CastMatch = namedtuple('CastMatch', 'role playedby cosine_dist')

def best_match(got_roles_list, clust_dict, vectorizer, topic_model, cluster_model):
    
    matches = []
    
    for person in got_roles_list:

        best_match = None
        best_fit = -np.inf
        
        words = vectorizer.transform([person['clean_doc']]).toarray()
        words_reduced = topic_model.transform(words)
        persons_cluster = cluster_model.predict(words_reduced)

        for cast in clust_dict[persons_cluster[0]]:

            cluster_person = [model_a.model.loc[cast]]
            topic_model_person = topic_model.transform(cluster_person)
            cos_dist = cosine_distance(words_reduced[0],topic_model_person[0])

            if cos_dist > best_fit:
                best_fit = cos_dist
                best_match = cast

        matches.append(CastMatch(role=person['name'], playedby=best_match, cosine_dist=best_fit))
    
    return matches