multi-task-gradient-boosting/MultiTaskUpdater.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 zebang.zhzb on 2018/6/22.
//

#ifndef MTREE_MULTITASKUPDATER_H
#define MTREE_MULTITASKUPDATER_H

#include "Updater.h"

class MultiTaskUpdater : public Updater {
 public:
  MultiTaskUpdater() {}
  ~MultiTaskUpdater() {}
  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_MULTITASKUPDATER_H