如何发现数据集中哪些特征具有预测性?

我正在为这里提供的数据集开发一个机器学习算法。

数据集有26列。大部分数据似乎无关紧要。我如何有效且快速地确定哪些特征是有趣的——哪些特征可以告诉我一个给定的URL是短暂的还是永恒的(这是数据集中的因变量)?是否有智能的、程序化的Scikit-learn方法来做到这一点,还是仅仅是将每个特征与因变量(’label’,第26列)进行图形化对比,看看哪些特征有影响?

肯定有更好的方法,对吗?

编辑:我找到了一些分类器的代码——我如何打印出这里给每个特征的权重?

import numpy as npimport matplotlib.pyplot as plt  from sklearn import metrics,preprocessing,cross_validation  from sklearn.feature_extraction.text import TfidfVectorizer  import sklearn.linear_model as lm  import pandas as p  loadData = lambda f: np.genfromtxt(open(f,'r'), delimiter=' ')  print "loading data.."  traindata = list(np.array(p.read_table('train.tsv'))[:,2])  testdata = list(np.array(p.read_table('test.tsv'))[:,2])  y = np.array(p.read_table('train.tsv'))[:,-1]  tfv = TfidfVectorizer(min_df=3,  max_features=None, strip_accents='unicode',          analyzer='word',token_pattern=r'\w{1,}',ngram_range=(1, 2), use_idf=1,smooth_idf=1,sublinear_tf=1)  rd = lm.LogisticRegression(penalty='l2', dual=True, tol=0.0001,                              C=1, fit_intercept=True, intercept_scaling=1.0,                              class_weight=None, random_state=None)  X_all = traindata + testdata  lentrain = len(traindata)  print "fitting pipeline"  tfv.fit(X_all)  print "transforming data"  X_all = tfv.transform(X_all)  X = X_all[:lentrain]  X_test = X_all[lentrain:]  print "20 Fold CV Score: ", np.mean(cross_validation.cross_val_score(rd, X, y, cv=20, scoring='roc_auc'))  print "training on full data"  rd.fit(X,y)  pred = rd.predict_proba(X_test)[:,1]  testfile = p.read_csv('test.tsv', sep="\t", na_values=['?'], index_col=1)  pred_df = p.DataFrame(pred, index=testfile.index, columns=['label'])  pred_df.to_csv('benchmark.csv')  print "submission file created.."

回答:

许多已拟合的scikit-learn估计器都有一个属性feature_importances_(除了线性模型,它们称之为coef_),其中包含某种特征权重。通常情况下,权重越高,特征对最终预测的贡献就越大,这可以解释为这些特征更具预测性。(*)

这些属性包含NumPy数组,形状为(n_features,),适用于二元分类、回归和非线性模型,或者(n_features, n_classes),适用于多类线性模型。

请参阅文档分类示例,了解如何使用这些属性。

(*) 所有关于过拟合的常见警告都适用:在不好的模型中,错误的特征可能会获得更高的权重。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注