wandb全名為Weights & Biases,用來幫助我們追蹤機器學習的項目,透過wandb可以記錄模型訓練過程中指標的變化情況以及超參的設置,還能夠將輸出的結果進行視覺化的比對,幫助我們更好的分析模型在訓練過程中的問題,同時我們也可以透過它來進行團隊協作
wandb會將訓練過程中的參數,上傳到伺服器上,然後透過登入wandb來進行即時過程模型訓練過程中參數和指標的變化
要由四大模組組成,分別是:
pip install wandb
wandb.login(key="填入你的API Keys")
##訓練模型
查看模型在測試集上Accuracy和loss的變化
看模型的預測效果
#查看訓練過程中系統參數(GPU和CPU等)的變化情況
###########以上是Wandb不可缺少的機器學習分析工具的詳細內容。更多資訊請關注PHP中文網其他相關文章!