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']])