跳到主要内容

递归特征消除

递归特征消除
**项目id：6227
https://scikit-learn.org/stable/auto_examples/feature_selection/plot_rfe_digits.html#sphx-glr-auto-examples-feature-selection-plot-rfe-digits-py 下图为项目总览：43.png 项目有四个节点组成：
数据加载：数据生成，生成用于训练的数据
模型定义：这里使用了SVC支持向量分类模型，参数为C=1，kernel=“linear”
递归特征消除组件设置：设置保留一个特征
结果展示

import os

import joblib
from matplotlib import pyplot as plt

import suanpan
from arguments import SklearnModel
from suanpan.app import app
from suanpan.app.arguments import File, Folder

TMP_FOLDER = "/tmp/result"


@app.input(File(key="inputModel1", name="model", type="model"))
@app.output(Folder(key="outputData1"))
def HelloWorld(context):
    args = context.args

    if not os.path.exists(TMP_FOLDER):
        os.makedirs(TMP_FOLDER)

    model = joblib.load(args.inputModel1)
    ranking = model.ranking_.reshape(8, 8)

    # Plot pixel ranking
    plt.matshow(ranking, cmap=plt.cm.Blues)
    plt.colorbar()
    plt.title("Ranking of pixels with RFE")

    plt.savefig(os.path.join(TMP_FOLDER, "result.png"), format="png")

    return TMP_FOLDER


if __name__ == "__main__":
    suanpan.run(app)

最终结果如下：44.png