Skip to content

Commit baa1480

Browse files
committed
Register loss and derivatives to Ensemble for easier API
1 parent ca5f813 commit baa1480

File tree

3 files changed

+259
-237
lines changed

3 files changed

+259
-237
lines changed

R-package/inst/include/ensemble.hpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,10 @@ class ENSEMBLE
3333
double get_extra_param();
3434
std::string get_loss_function();
3535

36+
double loss(Tvec<double> &y, Tvec<double> &pred, Tvec<double> &w);
37+
Tvec<double> dloss(Tvec<double> &y, Tvec<double> &pred);
38+
Tvec<double> ddloss(Tvec<double> &y, Tvec<double> &pred);
39+
3640
double initial_prediction(Tvec<double> &y, std::string loss_function, Tvec<double> &w);
3741
void train(Tvec<double> &y, Tmat<double> &X, int verbose, bool greedy_complexities,
3842
bool force_continued_learning, Tvec<double> &w, Tvec<double> &offset,

0 commit comments

Comments
 (0)