DTree cancer,pre-prune,post-prune

cancer # Decision Tree로 cancer 데이터를 학습한다. # ---------------------------------------- import matplotlib.pyplot as plt from sklearn.datasets import load_breast_cancer from sklearn.model_selection import train_test_split from sklearn.tree import DecisionTreeClassifier cancer = load_breast_cancer() # Train 데이터 세트와 Test 데이터 세트를 구성한다 trainX, testX, trainY, testY = \ train_test_split(cancer['data'], cancer['target'], test_size = 0.2) trainGini = [] testGini = [] trainEntropy = [] testEntropy = [] for k...

원문링크 : DTree cancer,pre-prune,post-prune