Skip to content

Commit

Permalink
rm inappropriate tests
Browse files Browse the repository at this point in the history
  • Loading branch information
ablaom committed Jan 28, 2021
1 parent 4c22c95 commit 6636de2
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions test/runtests.jl
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@ labels = features * weights;
features = MLJBase.table(features)
fitresultR, cacheR, reportR = MLJBase.fit(plain_regressor, 0, features, labels);
rpred = predict(plain_regressor, fitresultR, features);
info_dict(XGBoostRegressor)

plain_regressor.objective = "gamma"
labels = abs.(labels)
Expand Down Expand Up @@ -53,7 +52,6 @@ ycount = [rand(rng, Poisson(λᵢ)) for λᵢ ∈ λ]

fitresultC, cacheC, reportC = MLJBase.fit(count_regressor, 0, Xtable, ycount);
cpred = predict(count_regressor, fitresultC, Xtable);
info_dict(XGBoostCount)

importances = reportC.feature_importances

Expand Down Expand Up @@ -105,8 +103,6 @@ fitresult, cache, report = MLJBase.fit(plain_classifier, 0,
yhat = predict_mode(plain_classifier, fitresult, selectrows(X, test))
@test Set(MLJBase.classes(yhat[1])) == Set(MLJBase.classes(y[train][1]))

info_dict(XGBoostClassifier)

# serialization:
serializable_fitresult =
MLJBase.save("mymodel", plain_classifier, fitresult)
Expand Down

0 comments on commit 6636de2

Please sign in to comment.