# 保存为csv文件import numpy as np
y_pred = model_lgb.predict(test.drop('subscribe', axis=1))
result = pd.DataFrame({'id':y_id,'subscribe':y_pred.astype(np.int32)})
result_map ={0:'no',1:'yes'}
result['subscribe']= result['subscribe'].map(result_map)
result.to_csv("predict1.csv", index=False)
result['subscribe'].value_counts()
其他模型
逻辑回归
from sklearn.linear_model import LogisticRegression
model = LogisticRegression()# 实例化
model.fit(train.drop('subscribe', axis=1), train['subscribe'])
KNN
from sklearn.neighbors import KNeighborsClassifier
model = KNeighborsClassifier(n_neighbors=5)
model.fit(train.drop('subscribe', axis=1), train['subscribe'])
决策树
from sklearn.tree import DecisionTreeClassifier
model = DecisionTreeClassifier(criterion='entropy', max_depth=7, min_impurity_decrease=0.0)
model.fit(train.drop('subscribe', axis=1), train['subscribe'])