modelData.py 文件源码

def save_decision_tree(treePath, model, fold_idx, featNames):
    if not os.path.exists(treePath):
        os.makedirs(treePath)
    export_graphviz(model, out_file=treePath+'fold'+str(fold_idx)+'.dot', feature_names=featNames, filled=True, class_names=["absent","mild","moderate","severe"], proportion = True)