亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

如何在加載模型并使用一個類進行訓練時對類進行編碼器轉換

如何在加載模型并使用一個類進行訓練時對類進行編碼器轉換

翻閱古今 2022-04-23 17:16:39
我已經將模型保存為my_model.h5. 我的7課程是array(['Drums Beating', 'Machine Digging', 'Man Movement',      'Manual Digging', 'Manual Digging - Deeper (1.5 to 3 feets)',      'Normal', 'Tunneling'], dtype=object)現在我必須訓練一個只有一個類的模型(suppose 'drums beating')。所以我將用舊權重初始化新訓練模型的權重。因此,當我需要對標簽進行編碼(例如drums beating)時,如何對其進行編碼,使其具有([0,0,0,1,0,0,0])我之前訓練時的虛擬值。為了清楚起見,鼓聲的早期虛擬值是[0,0,0,1,0,0,0]。但是當我按如下方式加載編碼器時with open('/home/ANN_Unrolled_30_sample_7_class/ANN_UNrolled_sample_30_7_class.pkl', 'rb') as f:   encoder = pkl.load(f)并應用encode.transform,它[0]只是因為新訓練模型中只有一個類。我該怎么做才能得到以前的假人本身([0,0,0,1,0,0,0]]如果我過分強調或過分強調某事,請在評論中告訴我。
查看完整描述

2 回答

?
浮云間

TA貢獻1829條經驗 獲得超4個贊

請執行下列操作


with open('/home/ANN_Unrolled_30_sample_7_class/ANN_UNrolled_sample_30_7_class.pkl', "rb") as infile:

   encoder = pkl.load(infile)

temp = encoder.transform(your_required_classes)

num = len(encoder.classes_)

k = to_categorical(temp,num_classes=num)


查看完整回答
反對 回復 2022-04-23
?
叮當貓咪

TA貢獻1776條經驗 獲得超12個贊

在對整個訓練數據(7 個類)進行訓練后,您需要對編碼器進行腌制,就像對神經網絡所做的那樣。

最好的方法是有一個單獨的數據轉換管道,可以腌制。我建議使用 sklearn 的這個管道


查看完整回答
反對 回復 2022-04-23
  • 2 回答
  • 0 關注
  • 100 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號