.. _l-KNNImputer-num-tr-default--o17: KNNImputer - num-tr - default - ================================ Fitted on a problem type *num-tr* (see :func:`find_suitable_problem `), method `transform` matches output . :: KNNImputer() +-----------------------+----------+ | index | 0 | +=======================+==========+ | skl_nop | 1 | +-----------------------+----------+ | onx_size | 11720 | +-----------------------+----------+ | onx_nnodes | 37 | +-----------------------+----------+ | onx_ninits | 10 | +-----------------------+----------+ | onx_doc_string | | +-----------------------+----------+ | onx_ir_version | 8 | +-----------------------+----------+ | onx_domain | ai.onnx | +-----------------------+----------+ | onx_model_version | 0 | +-----------------------+----------+ | onx_producer_name | skl2onnx | +-----------------------+----------+ | onx_producer_version | 1.13.1 | +-----------------------+----------+ | onx_ | 17 | +-----------------------+----------+ | onx_ai.onnx.ml | 1 | +-----------------------+----------+ | onx_op_Cast | 6 | +-----------------------+----------+ | onx_op_Reshape | 1 | +-----------------------+----------+ | onx_subgraphs | 1 | +-----------------------+----------+ | onx_op_Identity | 2 | +-----------------------+----------+ | onx_op_Identity_optim | 1 | +-----------------------+----------+ | onx_subgraphs_optim | 1 | +-----------------------+----------+ | onx_size_optim | 11594 | +-----------------------+----------+ | onx_nnodes_optim | 35 | +-----------------------+----------+ | onx_ninits_optim | 10 | +-----------------------+----------+ | fit__fit_X.shape | (112, 4) | +-----------------------+----------+ .. gdot:: digraph{ size=7; ranksep=0.25; nodesep=0.05; orientation=portrait; X [shape=box color=red label="X\nfloat((0, 4))" fontsize=10]; variable [shape=box color=green label="variable\nfloat((0, 4))" fontsize=10]; Ar_ArrayFeatureExtractorcst [shape=box label="Ar_ArrayFeatureExtractorcst\nfloat32((4, 112))\n[[ 4.3017502e+00 5.1691985e+00 5.9683514e+00 5...." fontsize=10]; Sc_Scancst [shape=box label="Sc_Scancst\nfloat32((112, 4))\n[[ 4.3017502e+00 3.9453187e+00 9.2195314e-01 1...." fontsize=10]; Ma_MatMulcst [shape=box label="Ma_MatMulcst\nfloat32((4, 112))\n[[0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. 0. ..." fontsize=10]; Ma_MatMulcst1 [shape=box label="Ma_MatMulcst1\nfloat32((4, 112))\n[[1.8505054e+01 2.6720613e+01 3.5621220e+01 3.3977..." fontsize=10]; Ma_Maxcst [shape=box label="Ma_Maxcst\nfloat32((1,))\n[1.]" fontsize=10]; Ma_MatMulcst2 [shape=box label="Ma_MatMulcst2\nfloat32((4, 112))\n[[1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. 1. ..." fontsize=10]; Mu_Mulcst [shape=box label="Mu_Mulcst\nfloat32((1,))\n[4.]" fontsize=10]; To_TopKcst [shape=box label="To_TopKcst\nint64((1,))\n[5]" fontsize=10]; Re_Reshapecst [shape=box label="Re_Reshapecst\nint64((3,))\n[ 4 -1 5]" fontsize=10]; Re_ReduceSumcst [shape=box label="Re_ReduceSumcst\nint64((1,))\n[1]" fontsize=10]; Sh_shape0 [shape=box label="Sh_shape0" fontsize=10]; Sh_Shape [shape=box style="filled,rounded" color=orange label="Shape\n(Sh_Shape)" fontsize=10]; X -> Sh_Shape; Sh_Shape -> Sh_shape0; Is_Y0 [shape=box label="Is_Y0" fontsize=10]; Is_IsNaN [shape=box style="filled,rounded" color=orange label="IsNaN\n(Is_IsNaN)" fontsize=10]; X -> Is_IsNaN; Is_IsNaN -> Is_Y0; Ca_output0 [shape=box label="Ca_output0" fontsize=10]; Ca_Cast [shape=box style="filled,rounded" color=orange label="Cast\n(Ca_Cast)\nto=1" fontsize=10]; Is_Y0 -> Ca_Cast; Ca_Cast -> Ca_output0; Ca_output02 [shape=box label="Ca_output02" fontsize=10]; Ca_Cast1 [shape=box style="filled,rounded" color=orange label="Cast\n(Ca_Cast1)\nto=1" fontsize=10]; Is_Y0 -> Ca_Cast1; Ca_Cast1 -> Ca_output02; Co_output0 [shape=box label="Co_output0" fontsize=10]; Co_ConstantOfShape [shape=box style="filled,rounded" color=orange label="ConstantOfShape\n(Co_ConstantOfShape)\nvalue=[0.]" fontsize=10]; Sh_shape0 -> Co_ConstantOfShape; Co_ConstantOfShape -> Co_output0; Ma_Y02 [shape=box label="Ma_Y02" fontsize=10]; Ma_MatMul1 [shape=box style="filled,rounded" color=orange label="MatMul\n(Ma_MatMul1)" fontsize=10]; Ca_output0 -> Ma_MatMul1; Ma_MatMulcst1 -> Ma_MatMul1; Ma_MatMul1 -> Ma_Y02; Su_C02 [shape=box label="Su_C02" fontsize=10]; Su_Sub1 [shape=box style="filled,rounded" color=orange label="Sub\n(Su_Sub1)" fontsize=10]; Ma_Maxcst -> Su_Sub1; Ca_output02 -> Su_Sub1; Su_Sub1 -> Su_C02; Wh_output0 [shape=box label="Wh_output0" fontsize=10]; Wh_Where [shape=box style="filled,rounded" color=orange label="Where\n(Wh_Where)" fontsize=10]; Is_Y0 -> Wh_Where; Co_output0 -> Wh_Where; X -> Wh_Where; Wh_Where -> Wh_output0; Mu_C01 [shape=box label="Mu_C01" fontsize=10]; Mu_Mul [shape=box style="filled,rounded" color=orange label="Mul\n(Mu_Mul)" fontsize=10]; Wh_output0 -> Mu_Mul; Wh_output0 -> Mu_Mul; Mu_Mul -> Mu_C01; Ma_Y03 [shape=box label="Ma_Y03" fontsize=10]; Ma_MatMul2 [shape=box style="filled,rounded" color=orange label="MatMul\n(Ma_MatMul2)" fontsize=10]; Su_C02 -> Ma_MatMul2; Ma_MatMulcst2 -> Ma_MatMul2; Ma_MatMul2 -> Ma_Y03; UU014UU [shape=box label="UU014UU" fontsize=10]; UU015UU [shape=box label="UU015UU" fontsize=10]; subgraph cluster_Scan139904445564288_139908381009904 { label="Scan\n(Sc_Scan)\nbody=node {\n input: 'next_in'...\nnum_scan_inputs=1"; fontsize=10; color=black; B_next_in [shape=box color=red label="next_in\nfloat((0, 4))" fontsize=10]; B_next [shape=box color=red label="next\nfloat((0,))" fontsize=10]; B_next_out [shape=box color=green label="next_out\nfloat((0, 0))" fontsize=10]; B_scan_out [shape=box color=green label="scan_out\nfloat((0,))" fontsize=10]; B_cdistd_8_Identity [shape=box style="filled,rounded" color=orange label="Identity\n(cdistd_8_Identity)" fontsize=10]; B_next_in -> B_cdistd_8_Identity; B_cdistd_8_Identity -> B_next_out; B_cdistdf_8_C0 [shape=box label="cdistdf_8_C0" fontsize=10]; B_cdistdf_8_Sub [shape=box style="filled,rounded" color=orange label="Sub\n(cdistdf_8_Sub)" fontsize=10]; B_next_in -> B_cdistdf_8_Sub; B_next -> B_cdistdf_8_Sub; B_cdistdf_8_Sub -> B_cdistdf_8_C0; B_cdistdf_8_reduced0 [shape=box label="cdistdf_8_reduced0" fontsize=10]; B_cdistdf_8_ReduceSumSquare [shape=box style="filled,rounded" color=orange label="ReduceSumSquare\n(cdistdf_8_ReduceSumSquare)\naxes=[1]\nkeepdims=0" fontsize=10]; B_cdistdf_8_C0 -> B_cdistdf_8_ReduceSumSquare; B_cdistdf_8_ReduceSumSquare -> B_cdistdf_8_reduced0; B_cdistdf_8_Identity [shape=box style="filled,rounded" color=orange label="Identity\n(cdistdf_8_Identity)" fontsize=10]; B_cdistdf_8_reduced0 -> B_cdistdf_8_Identity; B_cdistdf_8_Identity -> B_scan_out; } Wh_output0 -> B_next_in; Sc_Scancst -> B_next; B_next_out -> UU014UU; B_scan_out -> UU015UU; Sc_Scan -> B_cdistd_8_Identity [lhead=cluster_Scan139904445564288_139908381009904]; Wh_output0 -> Sc_Scan; Sc_Scancst -> Sc_Scan; Sc_Scan -> UU014UU; Sc_Scan -> UU015UU; Ma_Y0 [shape=box label="Ma_Y0" fontsize=10]; Ma_MatMul [shape=box style="filled,rounded" color=orange label="MatMul\n(Ma_MatMul)" fontsize=10]; Mu_C01 -> Ma_MatMul; Ma_MatMulcst -> Ma_MatMul; Ma_MatMul -> Ma_Y0; Tr_transposed01 [shape=box label="Tr_transposed01" fontsize=10]; Tr_Transpose [shape=box style="filled,rounded" color=orange label="Transpose\n(Tr_Transpose)\nperm=[1 0]" fontsize=10]; UU015UU -> Tr_Transpose; Tr_Transpose -> Tr_transposed01; Ma_max0 [shape=box label="Ma_max0" fontsize=10]; Ma_Max [shape=box style="filled,rounded" color=orange label="Max\n(Ma_Max)" fontsize=10]; Ma_Maxcst -> Ma_Max; Ma_Y03 -> Ma_Max; Ma_Max -> Ma_max0; Ad_C0 [shape=box label="Ad_C0" fontsize=10]; Ad_Add [shape=box style="filled,rounded" color=orange label="Add\n(Ad_Add)" fontsize=10]; Ma_Y0 -> Ad_Add; Ma_Y02 -> Ad_Add; Ad_Add -> Ad_C0; Su_C0 [shape=box label="Su_C0" fontsize=10]; Su_Sub [shape=box style="filled,rounded" color=orange label="Sub\n(Su_Sub)" fontsize=10]; Tr_transposed01 -> Su_Sub; Ad_C0 -> Su_Sub; Su_Sub -> Su_C0; Di_C01 [shape=box label="Di_C01" fontsize=10]; Di_Div [shape=box style="filled,rounded" color=orange label="Div\n(Di_Div)" fontsize=10]; Su_C0 -> Di_Div; Ma_max0 -> Di_Div; Di_Div -> Di_C01; Mu_C0 [shape=box label="Mu_C0" fontsize=10]; Mu_Mul1 [shape=box style="filled,rounded" color=orange label="Mul\n(Mu_Mul1)" fontsize=10]; Di_C01 -> Mu_Mul1; Mu_Mulcst -> Mu_Mul1; Mu_Mul1 -> Mu_C0; To_Values0 [shape=box label="To_Values0" fontsize=10]; To_Indices1 [shape=box label="To_Indices1" fontsize=10]; To_TopK [shape=box style="filled,rounded" color=orange label="TopK\n(To_TopK)\nlargest=0\nsorted=1" fontsize=10]; Mu_C0 -> To_TopK; To_TopKcst -> To_TopK; To_TopK -> To_Values0; To_TopK -> To_Indices1; Fl_output0 [shape=box label="Fl_output0" fontsize=10]; Fl_Flatten [shape=box style="filled,rounded" color=orange label="Flatten\n(Fl_Flatten)" fontsize=10]; To_Indices1 -> Fl_Flatten; Fl_Flatten -> Fl_output0; Ar_Z0 [shape=box label="Ar_Z0" fontsize=10]; Ar_ArrayFeatureExtractor [shape=box style="filled,rounded" color=orange label="ArrayFeatureExtractor\n(Ar_ArrayFeatureExtractor)" fontsize=10]; Ar_ArrayFeatureExtractorcst -> Ar_ArrayFeatureExtractor; Fl_output0 -> Ar_ArrayFeatureExtractor; Ar_ArrayFeatureExtractor -> Ar_Z0; Re_reshaped0 [shape=box label="Re_reshaped0" fontsize=10]; Re_Reshape [shape=box style="filled,rounded" color=orange label="Reshape\n(Re_Reshape)\nallowzero=0" fontsize=10]; Ar_Z0 -> Re_Reshape; Re_Reshapecst -> Re_Reshape; Re_Reshape -> Re_reshaped0; Tr_transposed0 [shape=box label="Tr_transposed0" fontsize=10]; Tr_Transpose1 [shape=box style="filled,rounded" color=orange label="Transpose\n(Tr_Transpose1)\nperm=[1 2 0]" fontsize=10]; Re_reshaped0 -> Tr_Transpose1; Tr_Transpose1 -> Tr_transposed0; Re_reduced0 [shape=box label="Re_reduced0" fontsize=10]; Re_ReduceSum [shape=box style="filled,rounded" color=orange label="ReduceSum\n(Re_ReduceSum)\nkeepdims=0" fontsize=10]; Tr_transposed0 -> Re_ReduceSum; Re_ReduceSumcst -> Re_ReduceSum; Re_ReduceSum -> Re_reduced0; Ca_output04 [shape=box label="Ca_output04" fontsize=10]; Ca_Cast2 [shape=box style="filled,rounded" color=orange label="Cast\n(Ca_Cast2)\nto=9" fontsize=10]; Tr_transposed0 -> Ca_Cast2; Ca_Cast2 -> Ca_output04; Ca_output03 [shape=box label="Ca_output03" fontsize=10]; Ca_Cast3 [shape=box style="filled,rounded" color=orange label="Cast\n(Ca_Cast3)\nto=1" fontsize=10]; Ca_output04 -> Ca_Cast3; Ca_Cast3 -> Ca_output03; Re_reduced02 [shape=box label="Re_reduced02" fontsize=10]; Re_ReduceSum1 [shape=box style="filled,rounded" color=orange label="ReduceSum\n(Re_ReduceSum1)\nkeepdims=0" fontsize=10]; Ca_output03 -> Re_ReduceSum1; Re_ReduceSumcst -> Re_ReduceSum1; Re_ReduceSum1 -> Re_reduced02; Ca_output06 [shape=box label="Ca_output06" fontsize=10]; Ca_Cast4 [shape=box style="filled,rounded" color=orange label="Cast\n(Ca_Cast4)\nto=9" fontsize=10]; Re_reduced02 -> Ca_Cast4; Ca_Cast4 -> Ca_output06; No_Y0 [shape=box label="No_Y0" fontsize=10]; No_Not [shape=box style="filled,rounded" color=orange label="Not\n(No_Not)" fontsize=10]; Ca_output06 -> No_Not; No_Not -> No_Y0; Ca_output05 [shape=box label="Ca_output05" fontsize=10]; Ca_Cast5 [shape=box style="filled,rounded" color=orange label="Cast\n(Ca_Cast5)\nto=1" fontsize=10]; No_Y0 -> Ca_Cast5; Ca_Cast5 -> Ca_output05; Ad_C02 [shape=box label="Ad_C02" fontsize=10]; Ad_Add1 [shape=box style="filled,rounded" color=orange label="Add\n(Ad_Add1)" fontsize=10]; Re_reduced02 -> Ad_Add1; Ca_output05 -> Ad_Add1; Ad_Add1 -> Ad_C02; Di_C0 [shape=box label="Di_C0" fontsize=10]; Di_Div1 [shape=box style="filled,rounded" color=orange label="Div\n(Di_Div1)" fontsize=10]; Re_reduced0 -> Di_Div1; Ad_C02 -> Di_Div1; Di_Div1 -> Di_C0; Wh_Where1 [shape=box style="filled,rounded" color=orange label="Where\n(Wh_Where1)" fontsize=10]; Is_Y0 -> Wh_Where1; Di_C0 -> Wh_Where1; X -> Wh_Where1; Wh_Where1 -> variable; }