2021-02-28
閱讀量:
459
如何保存機(jī)器學(xué)習(xí)模型
問(wèn):
如何保存機(jī)器學(xué)習(xí)模型
答:
在機(jī)器學(xué)習(xí)中,在使用scikit學(xué)習(xí)庫(kù)時(shí),我們需要將訓(xùn)練好的模型保存在文件中并將其恢復(fù)以便重復(fù)使用它來(lái)將模型與其他模型進(jìn)行比較,以便在新數(shù)據(jù)上測(cè)試模型。保存數(shù)據(jù)稱為Serializaion,而恢復(fù)數(shù)據(jù)稱為反序列化。
此外,我們處理不同類型和大小的數(shù)據(jù)。有些數(shù)據(jù)集很容易訓(xùn)練,即它們花費(fèi)的時(shí)間較少,但是大小較大(超過(guò)1GB)的數(shù)據(jù)集在使用GPU時(shí)可能需要很長(zhǎng)時(shí)間才能在本地機(jī)器上進(jìn)行訓(xùn)練。當(dāng)我們?cè)谀硞€(gè)不同的項(xiàng)目中或稍后需要相同的訓(xùn)練數(shù)據(jù)時(shí),為了避免浪費(fèi)訓(xùn)練時(shí)間,存儲(chǔ)訓(xùn)練模型以便將來(lái)可以隨時(shí)使用。
我們可以通過(guò)兩種方式在scikit中保存模型:Pickle string和joblib






評(píng)論(0)


暫無(wú)數(shù)據(jù)
CDA考試動(dòng)態(tài)
CDA報(bào)考指南
推薦帖子
0條評(píng)論
0條評(píng)論
0條評(píng)論
0條評(píng)論