[refactor] move rbm.model → model.model

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-06-02 08:55:16 +02:00
co-authored by Claude Sonnet 4.6
parent b41274c7a6
commit 1954064a54
17 changed files with 19 additions and 19 deletions
+1 -1
View File
@@ -23,7 +23,7 @@
"import torchvision\n", "import torchvision\n",
"import torchvision.transforms as transforms\n", "import torchvision.transforms as transforms\n",
"import matplotlib.pyplot as plt\n", "import matplotlib.pyplot as plt\n",
"from rbm.model import Model\n", "from model.model import Model\n",
"from rbm.entity import Entity, EntityParams, TrainingParams\n", "from rbm.entity import Entity, EntityParams, TrainingParams\n",
"from rbm.matrix import Mat, np, rms_error_accu, sample_gaussian\n", "from rbm.matrix import Mat, np, rms_error_accu, sample_gaussian\n",
"from compat.torch import Optimizer\n", "from compat.torch import Optimizer\n",
+1 -1
View File
@@ -29,7 +29,7 @@
"import torchvision\n", "import torchvision\n",
"import torchvision.transforms as transforms\n", "import torchvision.transforms as transforms\n",
"import matplotlib.pyplot as plt\n", "import matplotlib.pyplot as plt\n",
"from rbm.model import Model\n", "from model.model import Model\n",
"from rbm.entity import Entity, EntityParams, TrainingParams\n", "from rbm.entity import Entity, EntityParams, TrainingParams\n",
"from rbm.matrix import Mat, np, rms_error_accu, sample_gaussian\n", "from rbm.matrix import Mat, np, rms_error_accu, sample_gaussian\n",
"from compat.torch import Optimizer\n", "from compat.torch import Optimizer\n",
+1 -1
View File
@@ -21,7 +21,7 @@
"import torchvision\n", "import torchvision\n",
"import torchvision.transforms as transforms\n", "import torchvision.transforms as transforms\n",
"import matplotlib.pyplot as plt\n", "import matplotlib.pyplot as plt\n",
"from rbm.model import Model\n", "from model.model import Model\n",
"from rbm.entity import Entity, EntityParams, TrainingParams\n", "from rbm.entity import Entity, EntityParams, TrainingParams\n",
"from rbm.matrix import Mat, np, rms_error_accu\n", "from rbm.matrix import Mat, np, rms_error_accu\n",
"from compat.torch import Optimizer\n", "from compat.torch import Optimizer\n",
+1 -1
View File
@@ -23,7 +23,7 @@
"import torchvision\n", "import torchvision\n",
"import torchvision.transforms as transforms\n", "import torchvision.transforms as transforms\n",
"import matplotlib.pyplot as plt\n", "import matplotlib.pyplot as plt\n",
"from rbm.model import Model\n", "from model.model import Model\n",
"from rbm.entity import Entity, EntityParams, TrainingParams\n", "from rbm.entity import Entity, EntityParams, TrainingParams\n",
"from rbm.matrix import Mat, np, rms_error_accu\n", "from rbm.matrix import Mat, np, rms_error_accu\n",
"from compat.torch import Optimizer\n", "from compat.torch import Optimizer\n",
+1 -1
View File
@@ -21,4 +21,4 @@ readme = "README.md"
[tool.setuptools] [tool.setuptools]
package-dir = {"" = "src"} package-dir = {"" = "src"}
packages = ["rbm", "image", "stack", "label", "compat"] packages = ["rbm", "image", "stack", "label", "compat", "model"]
+1 -1
View File
@@ -1,5 +1,5 @@
from rbm.matrix import Mat, np from rbm.matrix import Mat, np
from rbm.model import Model from model.model import Model
from rbm.entity import Entity, EntityParams, TrainingParams from rbm.entity import Entity, EntityParams, TrainingParams
import math import math
View File
+4 -4
View File
@@ -1,10 +1,10 @@
import os import os
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from .entity import Entity from rbm.entity import Entity
from .train import train from rbm.train import train
from .matrix import Mat, np from rbm.matrix import Mat, np
from .status import Status from rbm.status import Status
class Model(ABC): class Model(ABC):
known_classes = [Entity] known_classes = [Entity]
+1 -1
View File
@@ -1,7 +1,7 @@
import random import random
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from rbm.model import Model from model.model import Model
from rbm.entity import Entity, EntityParams, TrainingParams from rbm.entity import Entity, EntityParams, TrainingParams
from rbm.matrix import Mat, np from rbm.matrix import Mat, np
+1 -1
View File
@@ -1,7 +1,7 @@
import os import os
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from rbm.model import Model from model.model import Model
from rbm.entity import Entity, EntityParams, TrainingParams from rbm.entity import Entity, EntityParams, TrainingParams
from rbm.matrix import Mat, np, read_armadillo from rbm.matrix import Mat, np, read_armadillo
+1 -1
View File
@@ -3,7 +3,7 @@ import numpy
import cv2 import cv2
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from rbm.model import Model from model.model import Model
from rbm.entity import Entity, EntityParams, TrainingParams from rbm.entity import Entity, EntityParams, TrainingParams
from image.sub_image import SubImageExtract, normalize from image.sub_image import SubImageExtract, normalize
from rbm.matrix import Mat, np, convert from rbm.matrix import Mat, np, convert
+1 -1
View File
@@ -1,7 +1,7 @@
import random import random
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from rbm.model import Model from model.model import Model
from rbm.entity import Entity, EntityParams, TrainingParams from rbm.entity import Entity, EntityParams, TrainingParams
from image.sub_image import normalize from image.sub_image import normalize
from rbm.matrix import Mat, np from rbm.matrix import Mat, np
+1 -1
View File
@@ -1,7 +1,7 @@
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from rbm.model import Model from model.model import Model
from label.label import Label from label.label import Label
from rbm.entity import Entity, EntityParams, TrainingParams from rbm.entity import Entity, EntityParams, TrainingParams
from rbm.matrix import Mat, np from rbm.matrix import Mat, np
+1 -1
View File
@@ -1,6 +1,6 @@
import os import os
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from rbm.model import Model from model.model import Model
from rbm.entity import Entity, EntityParams, TrainingParams from rbm.entity import Entity, EntityParams, TrainingParams
from rbm.matrix import Mat, np, read_armadillo from rbm.matrix import Mat, np, read_armadillo
from rbm.train import train, Status from rbm.train import train, Status
+1 -1
View File
@@ -1,6 +1,6 @@
from rbm.matrix import Mat, np from rbm.matrix import Mat, np
from rbm.entity import Entity, EntityParams, TrainingParams from rbm.entity import Entity, EntityParams, TrainingParams
from rbm.model import Model from model.model import Model
from image.sub_image import normalize from image.sub_image import normalize
WORK_DIR = "../../results" WORK_DIR = "../../results"
+1 -1
View File
@@ -1,4 +1,4 @@
from rbm.model import Model from model.model import Model
from rbm.entity import Entity, EntityParams, TrainingParams from rbm.entity import Entity, EntityParams, TrainingParams
from rbm.matrix import Mat, np from rbm.matrix import Mat, np
+1 -1
View File
@@ -1,7 +1,7 @@
import os import os
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from rbm.model import Model from model.model import Model
from rbm.entity import Entity, EntityParams, TrainingParams from rbm.entity import Entity, EntityParams, TrainingParams
from rbm.matrix import Mat, np, read_armadillo from rbm.matrix import Mat, np, read_armadillo