feature selection for classifiers
dataset ---> mobile price dataset kaggle
import pandas as pd
import numpy as np
from sklearn.feature_selection import chi2
from sklearn.feature_selection import SelectKBest
df=pd.read_csv('train.csv')
df.head()
x_train = df.iloc[:,0:20]
y_train = df.iloc[:,-1]
best_features = SelectKBest(score_func=chi2,k=10)
fit= best_features.fit(x_train,y_train)
df_score = pd.DataFrame(fit.scores_)
dfcol=pd.DataFrame(x_train.columns)
Comments
Post a Comment