Open ghk829 opened 5 years ago
I run this code
import os
os.environ['is_test_suite']="True"
from auto_ml import Predictor
from auto_ml.utils import get_boston_dataset
from auto_ml.utils_models import load_ml_model
# Load data
df_train, df_test = get_boston_dataset()
# Tell auto_ml which column is 'output'
# Also note columns that aren't purely numerical
# Examples include ['nlp', 'date', 'categorical', 'ignore']
#
column_descriptions = {
'CHAS': 'output'
}
ml_predictor = Predictor(type_of_estimator='regressor', column_descriptions=column_descriptions)
ml_predictor.train(df_train,model_names=['GradientBoostingRegressor','RandomForestRegressor'])
I trained a model with parameters : model_names model_names=['GradientBoostingRegressor','RandomForestRegressor'] Then Error occured in