numpy和pytorch實現線性迴歸

一、numpy實現線性迴歸 輸入框19是定義的一個線性迴歸方法。 輸入框20是測試線性迴歸是否正常,輸入的數據權值分別是3,4,6,偏置項是5,設置學習率0.001,迭代次數5000,每迭代50次輸出一次損失函數的值。 驗證結果如下: 可以看到,損失函數的值在不斷減小,迴歸出來的各種參數也與我們的設定值幾乎相等。 二、pytorch實現線性迴歸 結果如下圖: 三、pytorch實現一個簡單的神經網
相關文章
相關標籤/搜索