Esempio n. 1
0
    train_similarity = Rules.load_similarity("train/similarity.txt")
    Rules.fill_dict(users, train_similarity)
    train_hottest = Rules.load_dict("train/hottest.txt")
    train_notbuy = Rules.load_dict("train/notbuy.txt")
    Rules.fill_dict(users, train_notbuy)
    train_topview = Rules.load_dict("train/topview.txt")
    Rules.fill_dict(users, train_topview)
    train_user_item = Rules.load_dict("train/user_item.txt")
    Rules.fill_dict(users, train_user_item)
    train_topk4 = Rules.load_dict("train/topk4.txt")
    train_user_label = Rules.load_dict("train/test_dict.txt")
    Rules.fill_dict(users, train_user_label)
    train_user_collect = Rules.load_dict("train/topk2.txt")
    Rules.fill_dict(users, train_user_collect)
    train_item_reco = Rules.load_dict("train/item_based_user_dict.txt")
    train_user_reco = Rules.Similarity(train_similarity, train_user_item,
                                       train_topk4)
    train_item_action = Rules.load_dict("train/user_action.txt")
    Rules.fill_dict(users, train_item_action)

    train_user_matrix, train_pool_dict, train_right_dict = Rules.classification(
        train_notbuy, train_hottest, train_topview, train_user_reco,
        train_item_reco, users, train_user_label, train_user_item,
        train_item_action)

    train_weights = EA.calculateWeights(train_hottest, train_notbuy,
                                        train_topview, train_item_reco,
                                        train_user_reco, train_user_label)
    train_result = EA.EArecommend(train_hottest, train_notbuy, train_topview,
                                  train_item_reco, train_user_reco,
                                  train_weights)
    EA.judge(train_result, train_user_label)