-
Notifications
You must be signed in to change notification settings - Fork 103
Expand file tree
/
Copy pathbuild.py
More file actions
39 lines (28 loc) · 1.06 KB
/
build.py
File metadata and controls
39 lines (28 loc) · 1.06 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# %load q02_best_k_features/build.py
# Default imports
import pandas as pd
data = pd.read_csv('data/house_prices_multivariate.csv')
from sklearn.feature_selection import SelectPercentile
from sklearn.feature_selection import f_regression
# Write your solution here:
def percentile_k_features(df,k=20):
X=df.iloc[:,:-1]
y=df.iloc[:,-1]
sp = SelectPercentile(f_regression,percentile=k)
sp.fit_transform(X,y)
features = X.columns.values[sp.get_support()]
scores = sp.scores_[sp.get_support()]
fs_score = list(zip(features,scores))
df = pd.DataFrame(fs_score,columns=['Name','Score'])
return df.sort_values(['Score','Name'],ascending = [False,True])['Name'].tolist()
X=data.iloc[:,:-1]
y=data.iloc[:,-1]
sp = SelectPercentile(f_regression,percentile=20)
sp.fit_transform(X,y)
features = X.columns.values[sp.get_support()]
scores = sp.scores_[sp.get_support()]
scores
fs_score = list(zip(features,scores))
df = pd.DataFrame(fs_score,columns=['Name','Score'])
df.head()
df.sort_values(['Score','Name'],ascending = [False,True])#['Name'].tolist()