为什么我的输出“ nan” nan＆quort'keras模型预测

Question

我正在尝试使AI试图预测质数序列的数字，但是我的模型输出“ [[NAN]]”。我的CSV文件的格式是这样的：

Prime，Prime的数量

，并且包含78498行遵循此模式。

我尝试查看模型，事实证明我的输入形状（无，1）。

有人知道如何解决这个问题吗？

这是我的代码：

import pandas as pd
import tensorflow as tf
from tensorflow import keras

data = pd.read_csv('primes.csv')
data = np.array(data, dtype=float)

data = data.T
numbers = np.array(data[0])
primes = np.array(data[1])

model = tf.keras.Sequential([keras.layers.Dense(units=1, input_shape=[1])])
model.compile(optimizer='sgd', loss='mean_squared_error')
model.fit(numbers, primes, epochs=10)

results = model.predict([1000000])

print(results)

model.save('model.h5')

我使用TensorFlow：2.8.0

Answer 1

这是一个示例转换类，您可以

在这种情况下使用分类值status_mapping

statusmapping={
    'OPEN': 0,
    'PENDING': 1,
    'PAYMENT_PENDING': 2,
            }
df['status_mapping'] = df['status'].map(statusmapping)
df

您可以将这些笔记本用于NAN值

https://www.kaggle.com/code/parulpandey/a-guide/a-guide-to to and-missing-missing-missing-values-values-values-values-values-values-values-values-values-values-values-values- python