Update app.py

app.py

@@ -13,6 +13,9 @@ from sklearn.ensemble import ExtraTreesRegressor
 filename = 'Dataset_RCS_3.csv'
 names0 = ['JET', "Suelo",'SPT', 'WtoC', 'Presion', 'Velocidad','RCS']
 dataset=pd.read_csv(filename, names=names0)
+# Dividimos el dataset en entrenamiento y test
+y = dataset['RCS']
+X = dataset.drop('RCS', axis=1)
 # Categorical data
 categorical_cols = ['JET', "Suelo"] 
 df = pd.get_dummies(X, columns = categorical_cols)