a ;ZшaUBу@s<dZddlmZmZddlmZddlmZddlZddl m Z ddlmZm Z ddlmZdd lmZmZmZdd lmZddlmZdd gd d gd dgddgddgddggZgdвZd dgd dgd dgddgddgddggZgdвZgdвZddДZddДZddДZddДZddДZ ddДZ!d d!ДZ"d"d#ДZ#d$d%ДZ$dS)&zG Testing for export functions of decision trees (sklearn.tree.export). щ)┌finditer┌search)┌dedent)┌RandomStateN)┌ is_classifier)┌DecisionTreeClassifier┌DecisionTreeRegressor)┌GradientBoostingClassifier)┌export_graphviz┌ plot_tree┌export_text)┌StringIO)┌NotFittedErrorщ■ щ щщ)rrrrrrщ)rrrчр?rr)rrrrrrc CsШtdddddН}|аttбt|ddН}d}||ks8JВt|ddgdd Н}d }||ksZJВt|ddgdd Н}d}||ks|JВt|ddddddddН}d}||ksдJВt|ddddН}d}||ks─JВt|dddddН}d}||ksцJВtdddddН}|jtttdН}t|ddddН}d}||kРs(JВtdddddН}|аttбt|dddddddН}d}||kРslJВtddН}|аttбt|ddd Н}d!}dS)"Nrr┌giniй┌ max_depth┌min_samples_split┌ criterion┌random_stateй┌out_fileaДdigraph Tree { node [shape=box, fontname="helvetica"] ; edge [fontname="helvetica"] ; 0 [label="X[0] <= 0.0\ngini = 0.5\nsamples = 6\nvalue = [3, 3]"] ; 1 [label="gini = 0.0\nsamples = 3\nvalue = [3, 0]"] ; 0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ; 2 [label="gini = 0.0\nsamples = 3\nvalue = [0, 3]"] ; 0 -> 2 [labeldistance=2.5, labelangle=-45, headlabel="False"] ; }Zfeature0Zfeature1)┌ feature_namesraИdigraph Tree { node [shape=box, fontname="helvetica"] ; edge [fontname="helvetica"] ; 0 [label="feature0 <= 0.0\ngini = 0.5\nsamples = 6\nvalue = [3, 3]"] ; 1 [label="gini = 0.0\nsamples = 3\nvalue = [3, 0]"] ; 0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ; 2 [label="gini = 0.0\nsamples = 3\nvalue = [0, 3]"] ; 0 -> 2 [labeldistance=2.5, labelangle=-45, headlabel="False"] ; }ZyesZno)┌class_namesraкdigraph Tree { node [shape=box, fontname="helvetica"] ; edge [fontname="helvetica"] ; 0 [label="X[0] <= 0.0\ngini = 0.5\nsamples = 6\nvalue = [3, 3]\nclass = yes"] ; 1 [label="gini = 0.0\nsamples = 3\nvalue = [3, 0]\nclass = yes"] ; 0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ; 2 [label="gini = 0.0\nsamples = 3\nvalue = [0, 3]\nclass = no"] ; 0 -> 2 [labeldistance=2.5, labelangle=-45, headlabel="False"] ; }TFZsans)┌filled┌impurity┌ proportionZspecial_characters┌roundedr┌fontnameaэdigraph Tree { node [shape=box, style="filled, rounded", color="black", fontname="sans"] ; edge [fontname="sans"] ; 0 [label=0 ≤ 0.0
samples = 100.0%
value = [0.5, 0.5]>, fillcolor="#ffffff"] ; 1 [label=value = [1.0, 0.0]>, fillcolor="#e58139"] ; 0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ; 2 [label=value = [0.0, 1.0]>, fillcolor="#399de5"] ; 0 -> 2 [labeldistance=2.5, labelangle=-45, headlabel="False"] ; }r)rrrzтdigraph Tree { node [shape=box, fontname="helvetica"] ; edge [fontname="helvetica"] ; 0 [label="X[0] <= 0.0\ngini = 0.5\nsamples = 6\nvalue = [3, 3]\nclass = y[0]"] ; 1 [label="(...)"] ; 0 -> 1 ; 2 [label="(...)"] ; 0 -> 2 ; })rrrZnode_idsa;digraph Tree { node [shape=box, style="filled", color="black", fontname="helvetica"] ; edge [fontname="helvetica"] ; 0 [label="node #0\nX[0] <= 0.0\ngini = 0.5\nsamples = 6\nvalue = [3, 3]", fillcolor="#ffffff"] ; 1 [label="(...)", fillcolor="#C0C0C0"] ; 0 -> 1 ; 2 [label="(...)", fillcolor="#C0C0C0"] ; 0 -> 2 ; })Z sample_weight)rr ra╧digraph Tree { node [shape=box, style="filled", color="black", fontname="helvetica"] ; edge [fontname="helvetica"] ; 0 [label="X[0] <= 0.0\nsamples = 6\nvalue = [[3.0, 1.5, 0.0]\n[3.0, 1.0, 0.5]]", fillcolor="#ffffff"] ; 1 [label="samples = 3\nvalue = [[3, 0, 0]\n[3, 0, 0]]", fillcolor="#e58139"] ; 0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ; 2 [label="X[0] <= 1.5\nsamples = 3\nvalue = [[0.0, 1.5, 0.0]\n[0.0, 1.0, 0.5]]", fillcolor="#f1bd97"] ; 0 -> 2 [labeldistance=2.5, labelangle=-45, headlabel="False"] ; 3 [label="samples = 2\nvalue = [[0, 1, 0]\n[0, 1, 0]]", fillcolor="#e58139"] ; 2 -> 3 ; 4 [label="samples = 1\nvalue = [[0.0, 0.5, 0.0]\n[0.0, 0.0, 0.5]]", fillcolor="#e58139"] ; 2 -> 4 ; }Z squared_error)rZleaves_parallelr┌rotater"r#aTdigraph Tree { node [shape=box, style="filled, rounded", color="black", fontname="sans"] ; graph [ranksep=equally, splines=polyline] ; edge [fontname="sans"] ; rankdir=LR ; 0 [label="X[0] <= 0.0\nsquared_error = 1.0\nsamples = 6\nvalue = 0.0", fillcolor="#f2c09c"] ; 1 [label="squared_error = 0.0\nsamples = 3\nvalue = -1.0", fillcolor="#ffffff"] ; 0 -> 1 [labeldistance=2.5, labelangle=-45, headlabel="True"] ; 2 [label="squared_error = 0.0\nsamples = 3\nvalue = 1.0", fillcolor="#e58139"] ; 0 -> 2 [labeldistance=2.5, labelangle=45, headlabel="False"] ; {rank=same ; 0} ; {rank=same ; 1; 2} ; }йr)rrz╛digraph Tree { node [shape=box, style="filled", color="black", fontname="helvetica"] ; edge [fontname="helvetica"] ; 0 [label="gini = 0.0\nsamples = 6\nvalue = 6.0", fillcolor="#ffffff"] ; }) r┌fit┌X┌yr ┌y2┌wr┌ y_degraded)┌clfZ contents1Z contents2йr-·=lib/python3.9/site-packages/sklearn/tree/tests/test_export.py┌test_graphviz_toysМ ° ∙ r/cCsфtdddН}tГ}tаtбПt||ГWdГn1s<0Y|аttбd}tjt |dНП t|ddgdНWdГn1sК0Yd}tjt |dНП"t|dgd вdНWdГn1s╬0Yd }tjt |dНП"t|аttбjГWdГn1Рs0YtГ}tаtбПt||gdНWdГn1РsT0YtГ}tjt ddНПt||d dНWdГn1РsШ0Ytjt ddНПt||ddНWdГn1Рs╓0YdS)Nrr)rrz?Length of feature_names, 1 does not match number of features, 2й┌match┌aйrz?Length of feature_names, 3 does not match number of features, 2)r2┌b┌czis not an estimator instance)rzshould be greater or equalr)┌ precisionzshould be an integer┌1) rr ┌pytest┌raisesrr r&r'r(┌ ValueError┌ TypeErrorZtree_┌ IndexError)r,┌out┌messager-r-r.┌test_graphviz_errors№s,(.02..r?cCsЖtdddН}|аttбtГ}t||dНtdddН}|аttб|jD]}t|d|dНqHtd|а бГD]}d|а бvslJВqldS)N┌friedman_mser)rrrr)Zn_estimatorsrz\[.*?samples.*?\])rr&r'r(r r r Zestimators_r┌getvalue┌group)r,┌dot_dataZ estimator┌findingr-r-r.┌test_friedman_mse_in_graphviz#s rEc Cs8tdГ}tdГ}t|аdб|аdбf|аdб|jdddНftdd d dНtd d dНfГD]▄\}}}|а||бd D]└}t|d|ddН}td|ГD]&}t t d|абГабГ|d ksОJВqОt|Гr─d}nd}t||ГD]&}t t d|абГабГ|d ks╥JВq╥td|ГD]*}t t d|абГабГ|d kРsJВРqqpqVdS)Nrщ)щr)щшщ)rG)rH)┌sizer@rr)rrrйrr)rIrT)rr6r!zvalue = \d+\.\d+z\.\d+zgini = \d+\.\d+zfriedman_mse = \d+\.\d+z<= \d+\.\d+) r┌zipZ random_sampleZrandintrrr&r r┌lenrrBr) Zrng_regZrng_clfr'r(r,r6rCrD┌patternr-r-r.┌test_precision2s2 №¤ $$rOcCstdddН}|аttбd}tjt|dНПt|ddНWdГn1sL0Yd}tjt|dНПt|d gd НWdГn1sМ0Yd}tjt|dНПt|ddНWdГn1s╩0Yd }tjt|dНПt|ddНWdГn1Рs 0YdS)NrrrKz max_depth bust be >= 0, given -1r0rr%z,feature_names must contain 2 elements, got 1r2r3zdecimals must be >= 0, given -1й┌decimalszspacing must be > 0, given 0йZspacing)rr&r'r(r8r9r:r)r,┌err_msgr-r-r.┌test_export_text_errors^s*,*rTcCsLtdddН}|аttбtdГаб}t|Г|ks4JВt|ddН|ksHJВt|ddН|ks\JВtdГаб}t|dd gd Н|ksАJВtdГаб}t|dd Н|ksаJВtdГаб}t|ddН|ks└JВddgddgddgddgddgddgddgg}gdв}tdddН}|а||бtdГаб}t|ddН|kРs0JВddgddgddgddgddgddgg}ddgddgddgddgddgddgg}tdddН}|а||бtdГаб}t|ddН|kРs║JВt|dddН|kРs╥JВdgdgdgdgdgdgg}tdddН}|а||бtdГаб}t|ddgdН|kРs,JВt|dddgdН|kРsHJВdS)NrrrKzh |--- feature_1 <= 0.00 | |--- class: -1 |--- feature_1 > 0.00 | |--- class: 1 r%щ zX |--- b <= 0.00 | |--- class: -1 |--- b > 0.00 | |--- class: 1 r2r4r3zФ |--- feature_1 <= 0.00 | |--- weights: [3.00, 0.00] class: -1 |--- feature_1 > 0.00 | |--- weights: [0.00, 3.00] class: 1 T)┌show_weightsz\ |- feature_1 <= 0.00 | |- class: -1 |- feature_1 > 0.00 | |- class: 1 rrRrr)rrrrrrrrIz{ |--- feature_1 <= 0.00 | |--- class: -1 |--- feature_1 > 0.00 | |--- truncated branch of depth 2 zy |--- feature_1 <= 0.0 | |--- value: [-1.0, -1.0] |--- feature_1 > 0.0 | |--- value: [1.0, 1.0] rP)rQrVzq |--- first <= 0.0 | |--- value: [-1.0, -1.0] |--- first > 0.0 | |--- value: [1.0, 1.0] ┌first)rQr)rQrVr)rr&r'r(r┌lstriprr)r,Zexpected_reportZX_lZy_lZX_moZy_moZregZX_singler-r-r.┌test_export_textps` . (( rYcCsАtdddddН}|аttбddg}t||dН}t|Гdks@JВ|dабd ksTJВ|d абdkshJВ|dабdks|JВdS) NrrZentropyr· first feat┌sepal_widthr3rz:first feat <= 0.0 entropy = 1.0 samples = 6 value = [3, 3]rz(entropy = 0.0 samples = 3 value = [3, 0]z(entropy = 0.0 samples = 3 value = [0, 3]йrr&r'r(rrMZget_textй┌pyplotr,rZnodesr-r-r.┌test_plot_tree_entropy╙s r_cCsАtdddddН}|аttбddg}t||dН}t|Гdks@JВ|dабd ksTJВ|d абdkshJВ|dабdks|JВdS) NrrrrrZr[r3rz7first feat <= 0.0 gini = 0.5 samples = 6 value = [3, 3]rz%gini = 0.0 samples = 3 value = [3, 0]z%gini = 0.0 samples = 3 value = [0, 3]r\r]r-r-r.┌test_plot_tree_giniчs r`cCs<tГ}tаtбПt|ГWdГn1s.0YdS)N)rr8r9rr)r^r,r-r-r.┌test_not_fitted_tree√sra)%┌__doc__┌rerr┌textwraprZnumpy.randomrr8Zsklearn.baserZsklearn.treerrZsklearn.ensembler r rr┌ior Zsklearn.exceptionsrr'r(r)r*r+r/r?rErOrTrYr_r`rar-r-r-r.┌s2((d',c