[refactor] move rbm.stack* + rbm.rnn_helper → stack/

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

src/stack/deep.py

@@ -1,7 +1,7 @@
-from .status import Status
-from .train import train
-from .stack import Stack, StackType
-from .matrix import Mat, np
+from rbm.status import Status
+from rbm.train import train
+from stack.stack import Stack, StackType
+from rbm.matrix import Mat, np
 
 class StackDeep(Stack):
 	def __init__(self, name: str, work_dir: str = '.'):
@@ -29,4 +29,4 @@ class StackDeep(Stack):
 	def pass_down_up(self, visible: Mat, from_layer_id: int = 0):
 		h = self.pass_up(visible, from_layer_id)
 		v = self.pass_down(h)
-		return v
+		return v

src/stack/factory.py

@@ -1,9 +1,9 @@
 import json
-from .stack import StackType
-from .layer import Layer
-from .entity import EntityParams, TrainingParams
-from .stack_deep import StackDeep
-from .stack_rnn import StackRnn
+from stack.stack import StackType
+from rbm.layer import Layer
+from rbm.entity import EntityParams, TrainingParams
+from stack.deep import StackDeep
+from stack.rnn import StackRnn
 
 class StackFactory:
 	@classmethod

src/stack/rnn.py

@@ -1,11 +1,10 @@
 import numpy as _np_cpu
-from .status import Status
-from .train import _to_gpu, cd_binary_binary, cd_gaussian_binary, cd_gaussian_gaussian, cd_binary_gaussian
-from .stack import Stack, StackType
-from .matrix import Mat, np, rms_error_accu, convert
-from .entity import Entity
-from .layer import Layer
-from .entity import EntityParams, TrainingParams
+from rbm.status import Status
+from rbm.train import _to_gpu, cd_binary_binary, cd_gaussian_binary, cd_gaussian_gaussian, cd_binary_gaussian
+from stack.stack import Stack, StackType
+from rbm.matrix import Mat, np, rms_error_accu, convert
+from rbm.entity import Entity, EntityParams, TrainingParams
+from rbm.layer import Layer
 
 _CD_FUNC = {
     Entity.Type.BB_RBM: cd_binary_binary,

src/stack/rnn_helper.py

@@ -38,4 +38,3 @@ def clamp_one_hot(src_dst: np.ndarray) -> np.ndarray:
 class Rnn:
 	def __init__(self):
 		pass
-

src/stack/stack.py

@@ -1,6 +1,6 @@
 import os
 from enum import Enum
-from .layer import Layer
+from rbm.layer import Layer
 
 class StackType(Enum):
 	Deep = "Deep",

src/tests/test_rbm.py

@@ -1,9 +1,9 @@
 import os.path
 import cv2 as cv
 from argparse import ArgumentParser
-from rbm.stack_factory import StackFactory
+from stack.factory import StackFactory
 from rbm.status import Status
-from rbm.stack_deep import StackDeep
+from stack.deep import StackDeep
 from rbm.matrix import Mat, np, convert, read_armadillo
 from rbm.entity import Entity
 

src/tests/test_rnn.py

@@ -1,6 +1,6 @@
 """Tests for StackRnn — shared-weights and unrolled (own-weights) modes."""
 import numpy as _np_cpu
-from rbm.stack_rnn import StackRnn
+from stack.rnn import StackRnn
 from rbm.matrix import np, convert
 from rbm.entity import EntityParams, TrainingParams
 from rbm.status import Status

src/tests/test_rnn_helper.py

@@ -1,6 +1,6 @@
 import numpy as np
 import pytest
-from rbm.rnn_helper import shift_right, shift_left, shift_up, shift_down, ch2idx, idx2ch, VOCAB
+from stack.rnn_helper import shift_right, shift_left, shift_up, shift_down, ch2idx, idx2ch, VOCAB
 
 M = np.array([
     [1, 2, 3],