When I run the code, I get the error shown in the attached picture.

How can I overcome this error? The code is the front part of an ANN analysis code. It gives an R2 and RMSE value as a result, but I doubt I’m doing something wrong. I wrote is as follows:

import pandas as pd
import numpy as np
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.metrics import r2_score, mean_squared_error
import tensorflow as tf
from tensorflow.models import Sequential
from tensorflow.layers import Dense, BatchNormalization, Dropout
from tensorflow.callbacks import ModelCheckpoint, EarlyStopping, ReduceLROnPlateau
from tensorflow.optimizers import Adam