feature_selection_final.py

from sklearn.svm import SVC
from sklearn.feature_selection import RFE
import pandas as pd
import time

# data = pd.read_csv('data/drug_cell/drug/17-AAG_train_data.csv')
# data = pd.read_csv('data/drug_cell/drug/Erlotinib_train_data.csv')
data = pd.read_csv('data/drug_cell/drug/Irinotecan_train_data.csv') # feature number选择9
data = data.fillna(0)
x = data.iloc[:, :-1]
y = data.iloc[:, -1]
svc = SVC(kernel="linear", C=1)
rfe = RFE(estimator=svc, n_features_to_select=9, step=1)
start = time.time()
rfe.fit(x, y)   # 训练时间特别长！！！
end = time.time()
print("特征选择运行时间：", end-start)
x_new = x[x.columns[rfe.get_support()]]
x_new['label'] = y
print(x_new)
x_new.to_csv('data/drug_cell/drug/Irinotecan_train_data-rfe.csv', index=False, float_format='%.2f')