Update app.py

app.py

@@ -5,9 +5,12 @@ from datasets import Dataset
 
 df=pd.read_csv("https://huggingface.co/spaces/Ralmao/Cart/blob/main/cart_data.csv", on_bad_lines='skip')
 dataset= Dataset.from_pandas(df)
-X = df.drop(['Cart_Abandoned'], axis = 1)
+# Select features (columns) for X
+X = df[['No_Checkout_Confirmed', 'No_Checkout_Initiated ', 'No_Customer_Login', 'Session_Activity_Count']]
+# Select target variable (column) for y
 y = df['Cart_Abandoned']
 
+
 #Importamos las librerias necesarias para la creacion del modelo
 from sklearn.model_selection import train_test_split
 
@@ -18,7 +21,7 @@ X_train, X_test, y_train, y_test = train_test_split(X,y, test_size=0.50, random_
 from sklearn.ensemble import RandomForestClassifier
 
 #Creacion del modelo
-random_forest = RandomForestClassifier(n_estimators=10, random_state=00000)
+random_forest = RandomForestClassifier(n_estimators=10)
 
 #Entrenamiento
 random_forest.fit(X_train,y_train)