Python pycharm learning code text record (8)
文章编号:Python学【2020】008号
日期:2019-2-19
文章为日常学习记录,以备温习使用,同时作为像我一样的新人学习使用,内容属于学习记录,版权归视频学习提供方。以每天追无聊小说的心态,每天坚持学习码代码。
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
music_data = pd.read_csv(‘music.csv’)
X = music_data.drop(“genre”,axis=1)
y = music_data[“genre”]
X_train,X_test,y_train,y_test = train_test_split(X,y,test_size = 0.2)
model = DecisionTreeClassifier()
model.fit(X_train,y_train)
predictions = model.predict(X_test)
score = accuracy_score(y_test,predictions)
score
Persisiting Model
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
from sklearn.model_selection import train_test_split
from sklearn.metrics import accuracy_score
from sklearn.externals import joblib
music_data = pd.read_csv('music.csv')
X = music_data.drop("genre",axis=1)
y = music_data["genre"]
X_train,X_test,y_train,y_test = train_test_split(X,y,test_size = 0.2)
model = DecisionTreeClassifier()
model.fit(X_train,y_train)
joblib.dump(model,'music-recommender.joblib')
[‘music-recommender.joblib’]
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
#Import dataset
music_data = pd.read_csv('music.csv')
X = music_data.drop("genre",axis=1)
y = music_data["genre"]
#Creat Model
model = DecisionTreeClassifier()
#Train Model
model.fit(X,y)
#Make predict
predictions = model.predict([[21,1],[22,0]])
predictions
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
#from sklearn.model_selection import train_test_split
#from sklearn.metrics import accuracy_score
from sklearn.externals import joblib
#import data set
music_data = pd.read_csv('music.csv')
X = music_data.drop("genre",axis=1)
y = music_data["genre"]
#X_train,X_test,y_train,y_test = train_test_split(X,y,test_size = 0.2)
#Create a model
model = DecisionTreeClassifier()
#Train it
model.fit(X_train,y_train)
joblib.dump(model,'music-recommender.joblib')
#predictions = model.predict([[21,1]])
#predictions
结果:在文件夹下创建文件“music-recommender.joblib”
增加技能快捷键技能
多行注释快捷键:选中要注释的段落,按下ctrl+/,
整段增加… …
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
model = joblib.load('music-recommender.joblib')
predictions = model.predict([[21,1]])
predictions
直接引入上段训练已存储的model文件,执行predict
Visualizing a Decision Tree