multi-task-gradient-boosting/SingleTaskUpdater.h at master · squall1988/multi-task-gradient-boosting · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
//
// Created by squall on 18-6-11.
//

#ifndef MTREE_SINGLETASKOBJ_H
#define MTREE_SINGLETASKOBJ_H

#include "Updater.h"

class SingleTaskUpdater : public Updater {
 public:
  SingleTaskUpdater() {}
  ~SingleTaskUpdater() {}
  float get_score(const vector<float> &feature,
                  const Matrix &gradients,
                  const vector<int> &sample_index,
                  float cut_point, float lambda);

  vector<float> get_scores(const vector<float> &feature,
                           const Matrix &gradients,
                           const vector<int> &task,
                           const int &task_num,
                           const vector<int> &sample_index,
                           float cut_point, float lambda);

};

#endif //MTREE_SINGLETASKOBJ_H