Tensorflow InvalidArgumentError: assertion failed: [Labels must be <= n_classes - 1]

Question

import pandas as pd
import matplotlib.pyplot as plt
import tensorflow as tf
df = pd.read_csv(\'pokemon_data.csv\')

df[\'Total\'] = df[\'HP\'] + df[\'Attack\'] + df[