o ╫tBhUBу@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}||ksJВt|ddgdd Н}d }||ks-JВt|ddgdd Н}d}||ks>JВt|ddddddddН}d}||ksRJВt|ddddН}d}||ksbJВt|dddddН}d}||kssJВtdddddН}|jtttdН}t|ddddН}d}||ksУJВtdddddН}|аttбt|dddddddН}d}||ks┤JВ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"] ; }┌feature0┌feature1)┌ 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"] ; }┌yes┌no)┌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"] ; }TF┌sans)┌filled┌impurity┌ proportion┌special_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)rr"rzт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 ; })rr$r┌node_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 ; })┌ sample_weight)r$r%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 ; }┌ squared_error)r$┌leaves_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)r$rz╛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)┌clf┌ contents1┌ contents2йr9·u/var/www/html/riverr-enterprise-integrations-main/venv/lib/python3.10/site-packages/sklearn/tree/tests/test_export.py┌test_graphviz_toysМ ° ∙ r;cCs▐tdddН}tГ}tаtбП t||ГWdГn1swY|аttбd}tjt |dНПt|ddgdНWdГn1sEwYd}tjt |dНПt|dgd вdНWdГn1sgwYd }tjt |dНПt|аttбjГWdГn1sЙwYtГ}tаtбПt||gdНWdГn1sиwYtГ}tjt ddНПt||d dНWdГn1s╔wYtjt ddНПt||ddНWdГdS1sшwYdS)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)r>┌b┌czis not an estimator instance)r"zshould be greater or equalr)┌ precisionzshould be an integer┌1) rr ┌pytest┌raisesrr r0r1r2┌ ValueError┌ TypeError┌tree_┌ IndexError)r6┌out┌messager9r9r:┌test_graphviz_errors№s: " rLcCsЖtdddН}|аttбtГ}t||dНtdddН}|аttб|jD] }t|d|dНq$td|а бГD] }d|а бvs@JВq6dS)N┌friedman_mser)rrrr)┌n_estimatorsrz\[.*?samples.*?\])rr0r1r2r r r ┌estimators_r┌getvalue┌group)r6┌dot_data┌ estimator┌findingr9r9r:┌test_friedman_mse_in_graphviz#s rUc Cs4tdГ}tdГ}t|аdб|аdбf|аdб|jdddНftdd d dНtd d dНfГD]l\}}}|а||бd D]^}t|d|ddН}td|ГD]}t t d|абГабГ|d ksZJВqGt|Гrbd}nd}t||ГD]}t t d|абГабГ|d ks|JВqitd|ГD]}t t d|абГабГ|d ksХJВqВq8q+dS)Nrщ)щr)щшщ)rW)rX)┌sizerMrr)rrrйrr)rYrT)rrBr&zvalue = \d+\.\d+z\.\d+zgini = \d+\.\d+zfriedman_mse = \d+\.\d+z<= \d+\.\d+) r┌zip┌ random_sample┌randintrrr0r r┌lenrrQr) ┌rng_reg┌rng_clfr1r2r6rBrRrT┌patternr9r9r:┌test_precision2s8 №¤ $$$ шЇrccCstdddН}|аttбd}tjt|dНПt|ddНWdГn1s&wYd}tjt|dНПt|d gd НWdГn1sFwYd}tjt|dНПt|ddНWdГn1sewYd }tjt|dНПt|ddНWdГdS1sЕwYdS)Nrrr[z max_depth bust be >= 0, given -1r<rr/z,feature_names must contain 2 elements, got 1r>r?zdecimals must be >= 0, given -1й┌decimalszspacing must be > 0, given 0й┌spacing)rr0r1r2rDrErFr)r6┌err_msgr9r9r:┌test_export_text_errors^s$ " ricCsFtdddН}|аttбtdГаб}t|Г|ksJВt|ddН|ks$JВt|ddН|ks.JВtdГаб}t|dd gd Н|ks@JВtdГаб}t|dd Н|ksPJВtdГаб}t|ddН|ks`JВddgddgddgddgddgddgddgg}gdв}tdddН}|а||бtdГаб}t|ddН|ksЧJВddgddgddgddgddgddgg}ddgddgddgddgddgddgg}tdddН}|а||бtdГаб}t|ddН|ks█JВt|dddН|ksцJВdgdgdgdgdgdgg}tdddН}|а||бtdГаб}t|ddgdН|kРsJВt|dddgdН|kРs!JВdS)Nrrr[zh |--- feature_1 <= 0.00 | |--- class: -1 |--- feature_1 > 0.00 | |--- class: 1 r/щ zX |--- b <= 0.00 | |--- class: -1 |--- b > 0.00 | |--- class: 1 r>r@r?zФ |--- 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 rrfrr)rrrrrrrrYz{ |--- 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] rd)rerkzq |--- first <= 0.0 | |--- value: [-1.0, -1.0] |--- first > 0.0 | |--- value: [1.0, 1.0] ┌first)rer)rerkr)rr0r1r2r┌lstriprr)r6┌expected_report┌X_l┌y_l┌X_mo┌y_mo┌reg┌X_singler9r9r:┌test_export_textpsz ∙ ∙ ∙ ∙. ∙(( ∙ ∙ rucCєАtdddddН}|аttбddg}t||dН}t|Гdks JВ|dабd ks*JВ|d абdks4JВ|dабdks>JВdS) Nrr┌entropyr· first feat┌sepal_widthr?rz: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]йrr0r1r2rr_┌get_textй┌pyplotr6r┌nodesr9r9r:┌test_plot_tree_entropy╙є rcCrv) Nrrrrrxryr?rz7first 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]rzr|r9r9r:┌test_plot_tree_giniчrАrБcCs>tГ}tаtбП t|ГWdГdS1swYdS)N)rrDrErr)r}r6r9r9r:┌test_not_fitted_tree√s " rВ)%┌__doc__┌rerr┌textwrapr┌numpy.randomrrD┌sklearn.baser┌sklearn.treerr┌sklearn.ensembler r rr┌ior ┌sklearn.exceptionsrr1r2r3r4r5r;rLrUrcrirurrБrВr9r9r9r:┌s4((d',c