target of RFR is Continuous not Finite

tlienart · tlienart · commit d9944ca07e37 · 2020-02-28T16:51:07.000+01:00
diff --git a/src/ScikitLearn/ensemble.jl b/src/ScikitLearn/ensemble.jl
@@ -194,7 +194,7 @@ MMI.fitted_params(model::RandomForestRegressor, (f, _, _)) = (
     )
 metadata_model(RandomForestRegressor,
     input   = Table(Count,Continuous),
-    target  = AbstractVector{<:Finite},
+    target  = AbstractVector{Continuous},
     weights = false,
     descr   = "Random forest regressor."
     )

Original file line number	Diff line number	Diff line change
`@@ -194,7 +194,7 @@ MMI.fitted_params(model::RandomForestRegressor, (f, _, _)) = (`
`194`	`194`	`)`
`195`	`195`	`metadata_model(RandomForestRegressor,`
`196`	`196`	`input = Table(Count,Continuous),`
`197`		`- target = AbstractVector{<:Finite},`
	`197`	`+ target = AbstractVector{Continuous},`
`198`	`198`	`weights = false,`
`199`	`199`	`descr = "Random forest regressor."`
`200`	`200`	`)`