示例#1
0
def rfecv_sklearn_example(image="rfecv_sklearn_example.png"):
    X, y = make_classification(
        n_samples=1000, n_features=25, n_informative=3, n_redundant=2,
        n_repeated=0, n_classes=8, n_clusters_per_class=1, random_state=0
    )

    _, ax = plt.subplots()

    oz = RFECV(SVC(kernel='linear', C=1), ax=ax)
    oz.fit(X, y)
    oz.poof(outpath=os.path.join(IMAGES, image))
示例#2
0
文件: rfecv.py 项目: minhto2802/mlviz
def rfecv_credit_example(image="rfecv_credit.png"):
    data = pd.read_csv(os.path.join(DATA, "credit", "credit.csv"))

    target = "default"
    features = [col for col in data.columns if col != target]

    X = data[features]
    y = data[target]

    _, ax = plt.subplots()
    cv = StratifiedKFold(5)
    oz = RFECV(RandomForestClassifier(), ax=ax, cv=cv, scoring='f1_weighted')
    oz.fit(X, y)
    oz.poof(outpath=os.path.join(IMAGES, image))
示例#3
0
def rfecv_credit_example(image="rfecv_credit.png"):
    data = pd.read_csv(os.path.join(DATA, "credit", "credit.csv"))

    target = "default"
    features = [col for col in data.columns if col != target]

    X = data[features]
    y = data[target]

    _, ax = plt.subplots()
    cv = StratifiedKFold(5)
    oz = RFECV(RandomForestClassifier(), ax=ax, cv=cv, scoring='f1_weighted')
    oz.fit(X, y)
    oz.poof(outpath=os.path.join(IMAGES, image))
示例#4
0
文件: rfecv.py 项目: minhto2802/mlviz
def rfecv_sklearn_example(image="rfecv_sklearn_example.png"):
    X, y = make_classification(n_samples=1000,
                               n_features=25,
                               n_informative=3,
                               n_redundant=2,
                               n_repeated=0,
                               n_classes=8,
                               n_clusters_per_class=1,
                               random_state=0)

    _, ax = plt.subplots()

    oz = RFECV(SVC(kernel='linear', C=1), ax=ax)
    oz.fit(X, y)
    oz.poof(outpath=os.path.join(IMAGES, image))
示例#5
0
def feature_selection(model, features, X, y):
    from sklearn.ensemble import RandomForestClassifier
    from sklearn.model_selection import StratifiedKFold
    from yellowbrick.features import RFECV

    # df = load_data('credit')

    # features = [col for col in data.columns if col != target]

    # X = data[features]
    # y = data[target]

    cv = StratifiedKFold(5)
    oz = RFECV(model, cv=cv, scoring='f1_weighted')

    oz.fit(X, y)
    oz.poof()
示例#6
0
def fazer_selecao_features_rfe(modelo):
    features = dados_completo.columns
    rfe = RFECV(modelo, cv=kfold, scoring=scoring)

    rfe.fit(dados_completo.drop(['classe'], axis=1),
            dados_completo['classe'].values)
    print(rfe.poof())
    print("Caraterísticas ordenadas pelo rank RFE:")
    print(sorted(zip(map(lambda x: round(x, 4), rfe.ranking_), features)))
    ranking = sorted(zip(rfe.support_, features))
    print("Características selecionadas", ranking)
    return rfe.transform(dados_completo.drop(['classe'], axis=1))
def fazer_selecao_features_rfe():
    features = X_treino.columns
    rfe = RFECV(RandomForestClassifier(random_state=random_state,
                                       oob_score=True),
                cv=kfold,
                scoring='accuracy')

    rfe.fit(X_treino, Y_treino.values.ravel())
    print(rfe.poof())
    print("Caraterísticas ordenadas pelo rank RFE:")
    print(sorted(zip(map(lambda x: round(x, 4), rfe.ranking_), features)))
    ranking = sorted(zip(rfe.support_, features))
    print("Características selecionadas", ranking)
    return rfe.transform(X_treino)
def fazer_selecao_features_rfe():
    features = X_treino.columns
    rfe = RFECV(RandomForestClassifier(random_state=random_state,
                                       oob_score=True,
                                       n_estimators=250,
                                       criterion='entropy',
                                       max_depth=75,
                                       max_features='log2',
                                       min_samples_leaf=1,
                                       min_samples_split=2),
                cv=kfold,
                scoring='accuracy')

    rfe.fit(X_treino, Y_treino.values.ravel())
    print(rfe.poof())
    print("Caraterísticas ordenadas pelo rank RFE:")
    print(sorted(zip(map(lambda x: round(x, 4), rfe.ranking_), features)))
    ranking = sorted(zip(rfe.support_, features))
    print("Características selecionadas", ranking)
    return rfe.transform(X_treino)
示例#9
0
cv = 3
Xt, Xv, yt, yv = \
    sklearn.model_selection.train_test_split(
        X, y, test_size=test_size, stratify=y, random_state=seed)
viz = RFECV(XGBClassifier(
    max_depth=7,
    n_estimators=1000,
    scale_pos_weight=50000,
    subsample=0.5,
    colsample_bylevel=0.5,
    colsample_bytree=0.5,
    seed=seed,
),
            cv=cv)
viz.fit(Xv, yv)
viz.poof()

# %%
feat = feat[viz.ranking_ < 5]
print(feat)
X = X[feat]

# %%
test_size = 0.33
cv = 3
Xt, Xv, yt, yv = \
    sklearn.model_selection.train_test_split(
        X, y, test_size=test_size, stratify=y, random_state=seed)
viz = RFECV(XGBClassifier(
    max_depth=7,
    n_estimators=1000,