From a8ce679361769ee531f82d64bc627a946dce66ec Mon Sep 17 00:00:00 2001
From: ankuru <ankuru.ahahahaha@gmail.com>
Date: Mon, 6 Jan 2025 12:51:16 +0900
Subject: [PATCH] =?UTF-8?q?feat:=20all=5Fstatus=E3=83=A1=E3=82=BD=E3=83=83?=
 =?UTF-8?q?=E3=83=89=E3=81=AE=E3=83=90=E3=82=B0=E4=BF=AE=E6=AD=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/model.cpp       | 4 ++--
 test/model_test.cpp | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/model.cpp b/src/model.cpp
index cabb634..b543356 100644
--- a/src/model.cpp
+++ b/src/model.cpp
@@ -114,8 +114,8 @@ double Model::kl_divergence(const Model& model){
 
 DataSet Model::all_status(){
   DataSet st;
-  rbm_utils::Binary binaryer(visible_dim);
-  for(int i=0; i<std::pow(visible_dim, 2); i++){
+  rbm_utils::Binary binaryer(visible_dim+hidden_dim);
+  for(int i=0; i<std::pow(2, visible_dim+hidden_dim); i++){
     st.push_back(binaryer.num2binary(i));
   }
   return st;
diff --git a/test/model_test.cpp b/test/model_test.cpp
index 7bead81..cbd2c87 100644
--- a/test/model_test.cpp
+++ b/test/model_test.cpp
@@ -112,10 +112,10 @@ TEST_F(ModelTest, CostFunc){
 
 TEST_F(ModelTest, AllStatus){
   DataSet status = rbm_model.all_status();
-  rbm_utils::Binary gen_bit(rbm_model.visible_dim);
+  rbm_utils::Binary gen_bit(rbm_model.visible_dim+rbm_model.hidden_dim);
 
   // std::cout << status.size() << std::endl;
-  for(auto i=0u; i<status.size(); i++){
+  for(auto i=0; i<64; i++){
     std::cout << status[i].transpose() << std::endl;
     ASSERT_EQ(status[i], gen_bit.num2binary(i));
   }