seed_user_search.py 文件源码

def train_decision_tree(file_name):
    file = open(file_name, 'r')
    train_data = json.load(file)
    file.close()
    train_list = list()
    train_result = list()
    for train_pair in train_data:
        tmp = formatting_data(train_pair[0])
        train_list.append(tmp)
        train_result.append(train_pair[1])
    my_clf = tree.DecisionTreeClassifier()
    my_clf.fit(train_list, train_result)
    return my_clf


# Setting up all the necessary preparation