From 946383d2bf8100ebd519f9eac4b9fb93fe0c7906 Mon Sep 17 00:00:00 2001
From: Relintai <relintai@protonmail.com>
Date: Tue, 14 Feb 2023 14:17:28 +0100
Subject: [PATCH] Small fixes.

---
 mlpp/log_reg/log_reg.cpp | 2 +-
 mlpp/log_reg/log_reg.h   | 7 +------
 2 files changed, 2 insertions(+), 7 deletions(-)

diff --git a/mlpp/log_reg/log_reg.cpp b/mlpp/log_reg/log_reg.cpp
index d99a638..2761e33 100644
--- a/mlpp/log_reg/log_reg.cpp
+++ b/mlpp/log_reg/log_reg.cpp
@@ -285,7 +285,7 @@ void MLPPLogReg::initialize() {
 	_initialized = true;
 }
 
-MLPPLogReg::MLPPLogReg(const Ref<MLPPMatrix> &p_input_set, const Ref<MLPPMatrix> &p_output_set, MLPPReg::RegularizationType p_reg, real_t p_lambda, real_t p_alpha) {
+MLPPLogReg::MLPPLogReg(const Ref<MLPPMatrix> &p_input_set, const Ref<MLPPVector> &p_output_set, MLPPReg::RegularizationType p_reg, real_t p_lambda, real_t p_alpha) {
 	_input_set = p_input_set;
 	_output_set = p_output_set;
 	_n = p_input_set->size().y;
diff --git a/mlpp/log_reg/log_reg.h b/mlpp/log_reg/log_reg.h
index c96385e..849ef78 100644
--- a/mlpp/log_reg/log_reg.h
+++ b/mlpp/log_reg/log_reg.h
@@ -17,9 +17,6 @@
 
 #include "../regularization/reg.h"
 
-#include <string>
-#include <vector>
-
 class MLPPLogReg : public Reference {
 	GDCLASS(MLPPLogReg, Reference);
 
@@ -56,9 +53,7 @@ public:
 	bool is_initialized();
 	void initialize();
 
-	//MLPPSoftmaxReg(const Ref<MLPPMatrix> &p_input_set, const Ref<MLPPMatrix> &p_output_set, MLPPReg::RegularizationType p_reg = MLPPReg::REGULARIZATION_TYPE_NONE, real_t p_lambda = 0.5, real_t p_alpha = 0.5);
-
-	MLPPLogReg(const Ref<MLPPMatrix> &p_input_set, const Ref<MLPPMatrix> &p_output_set, MLPPReg::RegularizationType p_reg = MLPPReg::REGULARIZATION_TYPE_NONE, real_t p_lambda = 0.5, real_t p_alpha = 0.5);
+	MLPPLogReg(const Ref<MLPPMatrix> &p_input_set, const Ref<MLPPVector> &p_output_set, MLPPReg::RegularizationType p_reg = MLPPReg::REGULARIZATION_TYPE_NONE, real_t p_lambda = 0.5, real_t p_alpha = 0.5);
 
 	MLPPLogReg();
 	~MLPPLogReg();