- 機器學習實戰:基于Sophon平臺的機器學習理論與實踐
- 星環科技人工智能平臺團隊
- 220字
- 2020-01-07 16:02:40
1.2.2 流程搭建
參考1.1.4節建立模型訓練與預測的流程(如圖1-4所示)。其中String Index和set role屬于數據預處理階段,分別對label設置index和label角色。然后進行數據切分,將數據切分為訓練集和驗證集,比例為8/2。接下來,利用人工神經網絡進行深度學習訓練。圖1-5展示了深度學習各層的配置。

圖1-4 mnist數據集:流程

圖1-5 mnist數據集:深度學習各層配置
最終模型訓練完畢,可以應用模型并在驗證集上驗證模型性能。圖1-6給出了混淆矩陣和召回率。關于Sophon機器學習平臺的詳細介紹請參見附錄A。

圖1-6 mnist數據集:模型性能