summaryrefslogtreecommitdiff
path: root/dtrain/test/logreg_cd/log_reg.h
blob: ecc560b85823118f37db44b0b9a575485810005e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
#ifndef _LOG_REG_H_
#define _LOG_REG_H_

#include <vector>
#include "sparse_vector.h"
#include "bin_class.h"

struct LogisticRegression : public Objective {
  double ObjectiveAndGradient(const SparseVector<double>& x,
                              const std::vector<TrainingInstance>& training_instances,
                              SparseVector<double>* g) const;
};

#endif