作者:煉己者
本博客所有內(nèi)容以學(xué)習(xí)、研究和分享為主,如需轉(zhuǎn)載,請(qǐng)聯(lián)系本人,標(biāo)明作者和出處,并且是非商業(yè)用途,謝謝!
用sklearn訓(xùn)練好我們的算法模型,以后肯定還會(huì)用到,所以我們要對(duì)它們進(jìn)行保存和讀取。方法有很多,我就講一種,用joblib進(jìn)行模型的讀寫(xiě)
# 訓(xùn)練模型
clf = svm.SVC(probability=True)
clf.fit(x,y)
# 給保存的模型的名字加上時(shí)間標(biāo)簽,以區(qū)分訓(xùn)練過(guò)程中產(chǎn)生的不同的模型
mdhms = time.strftime('%d%H%M', time.localtime(time.time()))
# 保存的模型的文件名
file = r'C:\Users\zhangyh4\Desktop\xietong1012\model\svm.joblib' + '_' + mdhms
# 保存模型
joblib.dump(clf,file)
# 讀取模型
svm_model = joblib.load(file)
以上便是模型的保存和讀取,大家有疑問(wèn)可以在底下評(píng)論交流