From 7b0d147aa351d8e13dd6692f91483b62145dd28b Mon Sep 17 00:00:00 2001
From: Relintai <relintai@protonmail.com>
Date: Thu, 28 Dec 2023 12:03:17 +0100
Subject: [PATCH] Fixed crashes and errors.

---
 mlpp/bernoulli_nb/bernoulli_nb.cpp     | 4 +++-
 mlpp/multinomial_nb/multinomial_nb.cpp | 3 +++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/mlpp/bernoulli_nb/bernoulli_nb.cpp b/mlpp/bernoulli_nb/bernoulli_nb.cpp
index 7333fe5..6c491aa 100644
--- a/mlpp/bernoulli_nb/bernoulli_nb.cpp
+++ b/mlpp/bernoulli_nb/bernoulli_nb.cpp
@@ -85,7 +85,9 @@ MLPPBernoulliNB::MLPPBernoulliNB(const Ref<MLPPMatrix> &p_input_set, const Ref<M
 	_prior_1 = 0;
 	_prior_0 = 0;
 
+	_vocab.instance();
 	_y_hat.instance();
+
 	_y_hat->resize(_output_set->size());
 
 	evaluate();
@@ -157,7 +159,7 @@ void MLPPBernoulliNB::evaluate() {
 
 		Vector<int> found_indices;
 
-		for (int j = 0; j < _input_set->size().y; j++) {
+		for (int j = 0; j < _input_set->size().x; j++) {
 			for (int k = 0; k < _vocab->size(); k++) {
 				if (_input_set->element_get(i, j) == _vocab->element_get(k)) {
 					score_0 += Math::log(static_cast<real_t>(_theta[0][_vocab->element_get(k)]));
diff --git a/mlpp/multinomial_nb/multinomial_nb.cpp b/mlpp/multinomial_nb/multinomial_nb.cpp
index a8c017e..58b8795 100644
--- a/mlpp/multinomial_nb/multinomial_nb.cpp
+++ b/mlpp/multinomial_nb/multinomial_nb.cpp
@@ -135,7 +135,10 @@ MLPPMultinomialNB::MLPPMultinomialNB(const Ref<MLPPMatrix> &p_input_set, const R
 	_output_set = p_output_set;
 	_class_num = pclass_num;
 
+	_priors.instance();
+	_vocab.instance();
 	_y_hat.instance();
+
 	_y_hat->resize(_output_set->size());
 
 	_initialized = true;