Good example but this line will error: model.fit(data, epochs=10) It needs to be split into x and y: model.fit(x_train, y_train, epochs=10)