text_plain_output_2.png text_plain_output_1.png
from sklearn.datasets import make_classification

import pandas as pd

NUM_FEATURES = 10
NUM_CLASSES = 3
NUM_POINTS = 100
MAX_DEPTH = 5
METRIC = 'gini'

x, y = make_classification(NUM_POINTS, NUM_FEATURES, n_informative=NUM_FEATURES // 2, n_classes=NUM_CLASSES)
df = pd.DataFrame(x)
df['y'] = y
df.head()