From 2c2f38d60b445ca830771d3e5a0eba080bd1ecd8 Mon Sep 17 00:00:00 2001
From: troyyyyy <49091847+troyyyyy@users.noreply.github.com>
Date: Thu, 24 Nov 2022 17:48:52 +0800
Subject: [PATCH] Update framework.py

---
 framework.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/framework.py b/framework.py
index ed013b7..2a1fdad 100644
--- a/framework.py
+++ b/framework.py
@@ -68,7 +68,7 @@ def train(model, abducer, train_data, test_data, epochs = 50, sample_num = -1, v
     
     # Set default parameters
     if sample_num == -1:
-        sample_num = len(X)
+        sample_num = len(train_X)
 
     if verbose < 1:
         verbose = epochs