mljar / mljar-supervised

Python package for AutoML on Tabular Data with Feature Engineering, Hyper-Parameters Tuning, Explanations and Automatic Documentation
https://mljar.com
MIT License
2.99k stars 400 forks source link

Error running the code in google collab #439

Open Rovky123 opened 3 years ago

Rovky123 commented 3 years ago

Here's the code import pandas as pd from sklearn.model_selection import train_test_split !pip install AutoML !pip install mljar-supervised from supervised.automl import AutoML

df = pd.read_csv( "https://raw.githubusercontent.com/Rovky123/Rovky123/main/1k40.csv", skipinitialspace=True, ) X_train, X_test, y_train, y_test = train_test_split( df[df.columns[:-1]], df["income"], test_size=0.25 )

automl = AutoML() automl.fit(X_train, y_train)

predictions = automl.predict(X_test)

I've found some errors loading AutoML in google collab. I want to solve the linear problem (1, 2,, 3, 4 and so on) and non linear problem (1, 2,1,2,1,2,1,2,1,2) I think that's the basic problem. Thanks in advance.

pplonski commented 3 years ago

To use mljar-supervised autoML in google colab please run following steps:

  1. Execute in the first cell: !pip install mljar-supervised
  2. After installation you will get the information that you need to restart runtime, please click the RESTART RUNTIME button: image
  3. After restart, remove install code and just import packages that you need. I've just checked and it works.