Skip to content

Commit adf7457

Browse files
committed
update 2 for SKL fit! routines for regression
1 parent 8824f03 commit adf7457

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/scikitlearnAPI.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ end
122122
[:pruning_purity_threshold, :min_samples_leaf, :nsubfeatures,
123123
:max_depth, :min_samples_split, :min_purity_increase, :rng])
124124

125-
function fit!{T<:Real}(dt::DecisionTreeRegressor, X::Matrix, y::Vector{T})
125+
function fit!(dt::DecisionTreeRegressor, X::Matrix, y::Vector)
126126
dt.root = build_tree(float.(y), X, dt.min_samples_leaf,
127127
dt.nsubfeatures, dt.max_depth, dt.min_samples_split, dt.min_purity_increase; rng=dt.rng)
128128
if dt.pruning_purity_threshold < 1.0
@@ -250,7 +250,7 @@ end
250250
# since it'll change throughout fitting, but it works
251251
:max_depth, :rng])
252252

253-
function fit!{T<:Real}(rf::RandomForestRegressor, X::Matrix, y::Vector{T})
253+
function fit!(rf::RandomForestRegressor, X::Matrix, y::Vector)
254254
rf.ensemble = build_forest(float.(y), X, rf.nsubfeatures,
255255
rf.ntrees, rf.min_samples_leaf, rf.partialsampling,
256256
rf.max_depth; rng=rf.rng)

0 commit comments

Comments
 (0)