例如,假設我正在構建一個 Transformer 模型(來自 Huggingface)并且訓練資料中有 20 個類,但是測驗資料中只有 5 個類。從huggingface中配置transformer模型,例如BertConfig,我們需要提供一個引數:num_labels。我應該將 num_labels 設定為 20 還是 5,因為測驗資料中只有 5 個類?
uj5u.com熱心網友回復:
20.因為你訓練的網路有 20 個類。
轉載請註明出處,本文鏈接:https://www.uj5u.com/ruanti/388060.html
