使用张量流制作了一个神经网络，该网络应该将 IP 与 7 种漏洞之一匹配，并给出 IP 具有的漏洞类型。

    model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(),
  tf.keras.layers.Dense(50, activation=tf.nn.relu),
  tf.keras.layers.Dense(7, activation=tf.nn.softmax)
])

model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])



model.fit(xs, ys, epochs=500)

执行此命令时的输出print(model.predict([181271844]))应该是从 1 到 7 的数字之一，但它给出的输出是

[[0.22288103 0.20282331 0.36847615 0.11339897 0.04456346 0.02391759 0.02393949]]

我似乎无法弄清楚问题是什么。