問題タブ [relu]

Keras を使用して訓練し、n 桁の 10 進数の i 番目の桁を予測しています。たとえば、入力が 123456 (つまり、n = 6) で i=2 の場合、プログラムは値 4 を見つける必要があります。入力を x_values に保存し、それに対応するトレーニング、テスト、および検証の値を保存します。 x_values の要素は y_values にあります。

私は2つのモデルを持っています。最初のものは単純な 'relu' で、スカラー入力を受け取り、それを 8 つの「ニューロン」に送ります。その後に、最後の層が 1 つのニューロンです。

2 番目のモデルは、16 ニューロン x 32 の深さです (最初のモデルよりかなり大きい)。

私の全体的なトレーニング セットには 15000 個のサンプルがあります。入力セットをトレーニング、テスト、検証セットに分割しました。

これらすべての後、損失と平均平均誤差を計算すると、両方とも非常に高くなっています (それぞれ約 8.3 と 2.5)。

この問題を解決する最善の方法は何でしょうか? ネットワークの幅、深さ、またはエポックの値をいじっても、目に見える効果はありませんでした。どんなヒントでも大歓迎です。