Skip to content

Commit 8824f03

Browse files
committed
minor cleanup to SKL show methods
1 parent 7b62aa6 commit 8824f03

File tree

1 file changed

+19
-19
lines changed

1 file changed

+19
-19
lines changed

src/scikitlearnAPI.jl

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -196,14 +196,14 @@ predict_proba(rf::RandomForestClassifier, X) =
196196

197197
predict(rf::RandomForestClassifier, X) = apply_forest(rf.ensemble, X)
198198

199-
function show(io::IO, dt::RandomForestClassifier)
199+
function show(io::IO, rf::RandomForestClassifier)
200200
println(io, "RandomForestClassifier")
201-
println(io, "ntrees: $(dt.ntrees)")
202-
println(io, "max_depth: $(dt.max_depth)")
203-
println(io, "nsubfeatures: $(dt.nsubfeatures)")
204-
println(io, "partialsampling: $(dt.partialsampling)")
205-
println(io, "classes: $(dt.classes)")
206-
println(io, "ensemble: $(dt.ensemble)")
201+
println(io, "ntrees: $(rf.ntrees)")
202+
println(io, "max_depth: $(rf.max_depth)")
203+
println(io, "nsubfeatures: $(rf.nsubfeatures)")
204+
println(io, "partialsampling: $(rf.partialsampling)")
205+
println(io, "classes: $(rf.classes)")
206+
println(io, "ensemble: $(rf.ensemble)")
207207
end
208208

209209
################################################################################
@@ -259,14 +259,14 @@ end
259259

260260
predict(rf::RandomForestRegressor, X) = apply_forest(rf.ensemble, X)
261261

262-
function show(io::IO, dt::RandomForestRegressor)
262+
function show(io::IO, rf::RandomForestRegressor)
263263
println(io, "RandomForestRegressor")
264-
println(io, "ntrees: $(dt.ntrees)")
265-
println(io, "max_depth: $(dt.max_depth)")
266-
println(io, "nsubfeatures: $(dt.nsubfeatures)")
267-
println(io, "partialsampling: $(dt.partialsampling)")
268-
println(io, "min_samples_leaf: $(dt.min_samples_leaf)")
269-
println(io, "ensemble: $(dt.ensemble)")
264+
println(io, "ntrees: $(rf.ntrees)")
265+
println(io, "max_depth: $(rf.max_depth)")
266+
println(io, "nsubfeatures: $(rf.nsubfeatures)")
267+
println(io, "partialsampling: $(rf.partialsampling)")
268+
println(io, "min_samples_leaf: $(rf.min_samples_leaf)")
269+
println(io, "ensemble: $(rf.ensemble)")
270270
end
271271

272272
################################################################################
@@ -311,12 +311,12 @@ predict(ada::AdaBoostStumpClassifier, X) =
311311
predict_proba(ada::AdaBoostStumpClassifier, X) =
312312
apply_adaboost_stumps_proba(ada.ensemble, ada.coeffs, X, ada.classes)
313313

314-
function show(io::IO, dt::AdaBoostStumpClassifier)
314+
function show(io::IO, ada::AdaBoostStumpClassifier)
315315
println(io, "AdaBoostStumpClassifier")
316-
println(io, "niterations: $(dt.niterations)")
317-
println(io, "coeffs: $(dt.coeffs)")
318-
println(io, "classes: $(dt.classes)")
319-
println(io, "ensemble: $(dt.ensemble)")
316+
println(io, "niterations: $(ada.niterations)")
317+
println(io, "coeffs: $(ada.coeffs)")
318+
println(io, "classes: $(ada.classes)")
319+
println(io, "ensemble: $(ada.ensemble)")
320320
end
321321

322322
################################################################################

0 commit comments

Comments
 (0)