20
Июл
2021

Не работает нейросеть на python keras

По гайду написал следующий код:

from numpy import loadtxt
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense

dataset = loadtxt('test.csv', delimiter=',')

X = dataset[:,0:2]
y = dataset[:,2]
print(X)
print(y)
model = Sequential()
model.add(Dense(2, activation='linear'))
model.add(Dense(1, activation='linear'))

model.compile(loss='mse', optimizer='sgd', metrics=['accuracy'])

model.fit(X, y, epochs=100)

model.predict(X)

CSV файл:

207,35,70.55
131,127,110.39
242,115.6,74.63
152,256,101.71
297,20,90.28

При обучении вижу такую картину:

1/1 [==============================] - 0s 453ms/step - loss: 117123.0625 - accuracy: 0.0000e+00
Epoch 2/100
1/1 [==============================] - 0s 0s/step - loss: 209229606702546944.0000 - accuracy: 0.0000e+00
Epoch 3/100
1/1 [==============================] - 0s 0s/step - loss: inf - accuracy: 0.0000e+00
Epoch 4/100
1/1 [==============================] - 0s 0s/step - loss: inf - accuracy: 0.0000e+00
Epoch 5/100
1/1 [==============================] - 0s 0s/step - loss: nan - accuracy: 0.0000e+00
Epoch 6/100
1/1 [==============================] - 0s 0s/step - loss: nan - accuracy: 0.0000e+00

Пробовал менять функции активации, оптимизатор, добавлял скрытые слои. Параметр loss менялся, но всё равно точность была 0. Что делать?

Источник: https://ru.stackoverflow.com/questions/1306937/%D0%9D%D0%B5-%D1%80%D0%B0%D0%B1%D0%BE%D1%82%D0%B0%D0%B5%D1%82-%D0%BD%D0%B5%D0%B9%D1%80%D0%BE%D1%81%D0%B5%D1%82%D1%8C-%D0%BD%D0%B0-python-keras

Тебе может это понравится...

Добавить комментарий