from sklearn.learning_curve import learning_curve
...

def runit():

...
    train_sizes, train_scores, test_scores = learning_curve(
        DecisionTreeClassifier(class_weight='balanced'), X, y, cv=5,
        train_sizes=train_sizes, n_jobs=4)

...
if __name__ == '__main__':
    runit()