@KFrank thanks for your time and answers.
You are right. It’s the first time I try to use regular variables instead of images on Pytorch.
If you could advise me on how to transform variables as input, I would be grateful.
So far I try (note I have 10 variables and 3 outputs):
X_train = np.array(X_train, dtype=np.float32)
X_train.shape

X_train = X_train.reshape(-1, 10)
X_train.shape

X_test = np.array(X_test, dtype=np.float32)
X_test.shape

X_test = X_test.reshape(-1, 10)
X_test.shape

Y_train = np.array(Y_train, dtype=np.float32)
y_train.shape

Y_train = Y_train.reshape(-1, 1)
Y_train.shape

I’d appreciate if you could let me know if my reasoning is correct.
Sincerely,