def test_groupby1(self): df = load_iris() train_out = decision_tree_regression_train( df, feature_cols=['sepal_length', 'sepal_width', 'petal_length'], label_col='petal_width', group_by=['species']) predict_out = decision_tree_regression_predict(df, train_out['model'])
def groupby1(self): df = get_iris() train_out = decision_tree_regression_train(df, feature_cols=['sepal_length', 'sepal_width', 'petal_length'], label_col='petal_width', group_by=['species']) predict_out = decision_tree_regression_predict(df, train_out['model']) print(predict_out['out_table'][['petal_width', 'prediction']])