summaryrefslogtreecommitdiff
path: root/dtrain/test/logreg/log_reg.h
diff options
context:
space:
mode:
Diffstat (limited to 'dtrain/test/logreg/log_reg.h')
-rw-r--r--dtrain/test/logreg/log_reg.h14
1 files changed, 14 insertions, 0 deletions
diff --git a/dtrain/test/logreg/log_reg.h b/dtrain/test/logreg/log_reg.h
new file mode 100644
index 00000000..ecc560b8
--- /dev/null
+++ b/dtrain/test/logreg/log_reg.h
@@ -0,0 +1,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