mjx-project / mjx

Mjx: A framework for Mahjong AI research
https://colab.research.google.com/drive/1m1wOT_K2YFtuV6IO7VgWk4ilVhTKqRFU?usp=sharing
MIT License
167 stars 18 forks source link

実験結果のまとめとnumpy配列への変換 #1148

Closed nissymori closed 1 year ago

nissymori commented 1 year ago

1127

nissymori commented 1 year ago

テストデータでの各局の絶対誤差の平均の比較結果です. no_logistic_afterは推論時にclipしています. 若干良くなっているのがわかります.

スクリーンショット 2022-10-16 22 36 17

最も結果が良かったモデルの重みをnumpy/にコミットしました.

nissymori commented 1 year ago

numpyの重みを使った予測と, jaxの重みを使った予測の差のテストようにjaxの重みもcommitしました.

test_inference.pyでテストができます.

sotetsuk commented 1 year ago

マージしちゃってください👍@nissymori

nissymori commented 1 year ago

了解です!!

nissymori commented 1 year ago

了解です!!