@@ -79,8 +79,8 @@ function fitted_machine(model, data...; throw=false, verbosity=1)
79
79
machine (model, data... )
80
80
fit! (mach, verbosity= - 1 )
81
81
train, _ = MLJBase. partition (1 : MLJBase. nrows (first (data)), 0.5 )
82
- fit! (mach, rows= train, verbosity= - 1 )
83
- fit! (mach, rows= :, verbosity= - 1 )
82
+ model isa Static || fit! (mach, rows= train, verbosity= - 1 )
83
+ model isa Static || fit! (mach, rows= :, verbosity= - 1 )
84
84
MLJBase. report (mach)
85
85
MLJBase. fitted_params (mach)
86
86
mach
90
90
function operations (fitted_machine, data... ; throw= false , verbosity= 1 )
91
91
message = " [:operations] Calling `predict`, `transform` and/or `inverse_transform` "
92
92
attempt (finalize (message, verbosity); throw) do
93
+ model = fitted_machine. model
93
94
operations = String[]
94
95
methods = MLJBase. implemented_methods (fitted_machine. model)
95
96
_, test = MLJBase. partition (1 : MLJBase. nrows (first (data)), 0.5 )
96
97
if :predict in methods
97
98
predict (fitted_machine, first (data))
98
- predict (fitted_machine, rows= test)
99
- predict (fitted_machine, rows= :)
99
+ model isa Static || predict (fitted_machine, rows= test)
100
+ model isa Static || predict (fitted_machine, rows= :)
100
101
push! (operations, " predict" )
101
102
end
102
103
if :transform in methods
103
104
W = transform (fitted_machine, first (data))
104
- transform (fitted_machine, rows= test)
105
- transform (fitted_machine, rows= :)
105
+ model isa Static || transform (fitted_machine, rows= test)
106
+ model isa Static || transform (fitted_machine, rows= :)
106
107
push! (operations, " transform" )
107
108
if :inverse_transform in methods
108
109
inverse_transform (fitted_machine, W)
0 commit comments