Skip to content

Commit 6388588

Browse files
committed
fix 1.6 error
1 parent 021aa6f commit 6388588

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

src/classification/main.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ function build_stump(
5656
weights = nothing;
5757
rng = Random.GLOBAL_RNG) where {S, T}
5858

59+
rng = mk_rng(rng)::Random.AbstractRNG
5960
t = treeclassifier.fit(
6061
X = features,
6162
Y = labels,

src/measures.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,7 @@ function _nfoldCV(classifier::Symbol, labels::AbstractVector{T}, features::Abstr
135135
predictions = apply_forest(model, test_features)
136136
elseif classifier == :stumps
137137
model, coeffs = build_adaboost_stumps(
138-
train_labels, train_features, n_iterations)
138+
train_labels, train_features, n_iterations, rng=rng)
139139
predictions = apply_adaboost_stumps(model, coeffs, test_features)
140140
end
141141
cm = confusion_matrix(test_labels, predictions)

0 commit comments

Comments
 (0)