使用logistic 回归解决良性、恶性肿瘤的二分类问题

准确率：0.9371428571428572

#!/usr/bin/python
# -*- coding:utf-8 -*-

import pandas as pd

df_train = pd.read_csv('../Datasets/Breast-Cancer/breast-cancer-train.csv')

df_test = pd.read_csv('../Datasets/Breast-Cancer/breast-cancer-test.csv')

from sklearn.linear_model import LogisticRegression

lr = LogisticRegression()

lr.fit(df_train[['Clump Thickness', 'Cell Size']], df_train['Type'])
print('Testing accuracy (all training samples):', lr.score(df_test[['Clump Thickness', 'Cell Size']], df_test['Type']))