第2关:癌症诊断
encoding=utf8
from sklearn.linear_model import LogisticRegression
def cancer_clf(train_data,train_label,test_data):
'''
train_data(ndarray):训练数据
train_label(ndarray):训练标签
test_data(ndarray):ces数据
'''
#********* Begin #
logreg = LogisticRegression(C=100)
logreg.fit(train_data, train_label)
predict = logreg.predict(test_data)
# End *********#
return predict