Пример #1
0
def task2a():
    DataHandler.vectors()
    DataHandler.createDictionaries1()

    actor_actor_sim_df, actorList = DataHandler.actor_actor_similarity_matrix()
    u, sigma, vt = decompositions.SVDDecomposition(actor_actor_sim_df, 3)
    semantics = np.matrix(vt).tolist()

    actorIdActorsDf = DataHandler.actor_info_df
    actorsInDf = list(actor_actor_sim_df.index)
    print("Top 3 semantics are:")
    for semantic in semantics:
        prettyPrintActorVector(semantic, actorsInDf, actorIdActorsDf)
        print("")

    split_group_with_index = formatter.splitGroup(u, 3)

    print("The three groupings are:")
    groups = tasksBusiness.get_partition_on_ids(split_group_with_index,
                                                actorIdActorsDf['name'])
    for x, v in groups.items():
        print('Group ' + str(x + 1) + ' : ' + str(v))
        print(" ")