Keras:训练模型时出现“损失:nan - 准确度:0.0000e+00”

时间:2021-05-25 04:14:41

标签: python tensorflow machine-learning keras

我正在制作一个模型(来自这个数据集:https://www.kaggle.com/karangadiya/fifa19),根据球员的一些统计数据猜测他的“整体”。我已经完成了所需的数据清理,但损失和准确度是“损失:nan - 准确度:0.0000e+00”。
以下是我的数据清理代码:-

%matplotlib notebook
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np

import tensorflow as tf
from tensorflow import keras
from sklearn.model_selection import train_test_split

data = pd.read_csv("./data.csv")
data.dropna().replace(np.nan, 0)
data.columns = [x.lower() for x in data.columns]
data = data.select_dtypes(include="number").drop(columns = ["id", "unnamed: 0"], axis=1)

y = data["overall"]
X = data[["crossing", "headingaccuracy", "standingtackle", "gkreflexes"]]

X_train, X_test, y_train, y_test = train_test_split(X,y, random_state=45)

X_train = (X_train - np.max(X_train))/(np.max(X_train) - np.min(X_train))
y_train = np.array(y_train, dtype = 'float32')

这是我的模型代码:-

model = keras.models.Sequential([
    keras.layers.Dense(4, input_shape=(4,)),
    keras.layers.Dense(128, activation=tf.nn.relu),
    keras.layers.Dense(1, activation=tf.nn.relu)
])
model.compile(optimizer="sgd", loss="mean_squared_error", metrics=["accuracy"])
model.fit(X_train, y_train, epochs=5)

这是模型训练时的输出:-

Epoch 1/5
427/427 [==============================] - 1s 2ms/step - loss: nan - accuracy: 0.0000e+00
Epoch 2/5
427/427 [==============================] - 1s 1ms/step - loss: nan - accuracy: 0.0000e+00
Epoch 3/5
427/427 [==============================] - 1s 1ms/step - loss: nan - accuracy: 0.0000e+00
Epoch 4/5
427/427 [==============================] - 1s 2ms/step - loss: nan - accuracy: 0.0000e+00
Epoch 5/5
427/427 [==============================] - 1s 2ms/step - loss: nan - accuracy: 0.0000e+00
<tensorflow.python.keras.callbacks.History at 0x13b16c6e280>

感谢您的帮助!

0 个答案:

没有答案