training.py 文件源码

def build_random_forest_model(x_train, y_train):

    rf_model = RandomForestClassifier(random_state=42)
    rf_model.fit(x_train, y_train.ravel())
    print "10-fold Cross validation score is :"
    print np.mean(cross_val_score(rf_model, x_train, y_train, cv=10))
    return rf_model