model.py 文件源码

def et_opt1(df_cell_train_feats, y_train, df_cell_test_feats):
    logging.info("train et_opt1 model")
    clf = ExtraTreesClassifier(n_estimators=500, n_jobs=-1, max_features="log2", min_samples_split=5, min_samples_leaf=1)
    clf.fit(df_cell_train_feats, y_train)
    y_test_pred = clf.predict_proba(df_cell_test_feats)
    return y_test_pred