ml5js/ml5-library

Default Learning Rates for Optimizers are higher than those in Tensorflow.js/Keras

Open

#1,249 建立於 2021年11月13日

在 GitHub 查看
 (2 留言) (0 反應) (0 負責人)JavaScript (6,136 star) (905 fork)batch import
bughelp wantedneural-network

描述

Dear ml5 community,

I'm submitting a new issue. Please see the details below.

→ Step 1: Describe the issue 📝

The default Optimizer learning rates are significantly higher than those in tensorflow.js. This can lead to sub-optimal accuracy rates when training.

→ Step 2: Screenshots or Relevant Documentation 🖼

Default ML5.js Regression Adam Optimizer learning rate: .20 Default Tensorflow.js Adam Optimizer learning rate: .001

貢獻者指南

技術棧
javascript
領域
machine learning
議題類型
bug
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
2
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
under 1 hour
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
stale
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
clear
前置要求
JavaScriptbasic knowledge of neural network optimizers
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
70
研究方向
Change the default learning rate for Adam optimizer from 0.2 to 0.001 in src/NeuralNetwork/index.js line 18 to match TensorFlow.js defaults. Ensure other optimizers have consistent defaults. Update any related tests.