机器学习CV代码练习（一）之逻辑回归分类垃圾短信

垃圾短信分类

• 垃圾短信数据集SMSSpamCollection.txt：链接：https://pan.baidu.com/s/13cM7xjO6uk4gCbjhfclGQA
  提取码：mzt5

from sklearn.linear_model import LogisticRegression
import numpy as np
import pandas as pd
from sklearn.feature_extraction.text import TfidfVectorizer
from sklearn.linear_model import LogisticRegression

df = pd.read_csv('SMSSpamCollection.txt',delimiter='\t',header=None)
y,x_train = df[0],df[1]
vectorize = TfidfVectorizer()#向量化
x = vectorize.fit_transform(x_train)
# print(x)
lr = LogisticRegression()
lr.fit(x,y)

test = vectorize.fit_transform(['I am a good boy.','Yes'])
predict = lr.predict(test)
print(predict)
# print(x_train,"   ",y)