源码库:
https://github.com/RGF-team/rgf/releases
rgf包的安装:
注意python2 及python3的包有区别,最新的rgf-python版本不支持python2,请安装之前的python版本,支撑python2的最高版本是3.9
pip install rgf-python==3.4.0
RGF源码如下:
Examples
from sklearn import datasets from sklearn.utils.validation import check_random_state from sklearn.model_selection import StratifiedKFold, cross_val_score from rgf.sklearn import RGFClassifier iris = datasets.load_iris() rng = check_random_state(0) perm = rng.permutation(iris.target.size) iris.data = iris.data[perm] iris.target = iris.target[perm] rgf = RGFClassifier(max_leaf=400, algorithm="RGF_Sib", test_interval=100, verbose=True) n_folds = 3 rgf_scores = cross_val_score(rgf, iris.data, iris.target, cv=StratifiedKFold(n_folds)) rgf_score = sum(rgf_scores)/n_folds print('RGF Classifier score: {0:.5f}'.format(rgf_score))
页面下拉,注意支持的版本