boltzmann9 0.1.3__py3-none-any.whl → 0.1.6__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- boltzmann9/cli.py +18 -18
- boltzmann9/model.py +1 -1
- {boltzmann9-0.1.3.dist-info → boltzmann9-0.1.6.dist-info}/METADATA +1 -1
- {boltzmann9-0.1.3.dist-info → boltzmann9-0.1.6.dist-info}/RECORD +7 -7
- {boltzmann9-0.1.3.dist-info → boltzmann9-0.1.6.dist-info}/WHEEL +0 -0
- {boltzmann9-0.1.3.dist-info → boltzmann9-0.1.6.dist-info}/entry_points.txt +0 -0
- {boltzmann9-0.1.3.dist-info → boltzmann9-0.1.6.dist-info}/top_level.txt +0 -0
boltzmann9/cli.py
CHANGED
|
@@ -8,7 +8,7 @@ from pathlib import Path
|
|
|
8
8
|
|
|
9
9
|
def cmd_new_project(args):
|
|
10
10
|
"""Create a new project."""
|
|
11
|
-
from
|
|
11
|
+
from boltzmann9.project import create_project
|
|
12
12
|
|
|
13
13
|
try:
|
|
14
14
|
create_project(args.project_name)
|
|
@@ -20,10 +20,10 @@ def cmd_new_project(args):
|
|
|
20
20
|
|
|
21
21
|
def cmd_train(args):
|
|
22
22
|
"""Train a model."""
|
|
23
|
-
from
|
|
24
|
-
from
|
|
25
|
-
from
|
|
26
|
-
from
|
|
23
|
+
from boltzmann9.config import load_config
|
|
24
|
+
from boltzmann9.pipeline import train_rbm, save_model
|
|
25
|
+
from boltzmann9.project import create_run, save_run_config, get_run_paths
|
|
26
|
+
from boltzmann9.run_utils import RunLogger, save_history, save_plots
|
|
27
27
|
|
|
28
28
|
project_path = Path(args.project)
|
|
29
29
|
|
|
@@ -57,9 +57,9 @@ def cmd_train(args):
|
|
|
57
57
|
cfg["data"]["csv_path"] = str(project_path / data_path)
|
|
58
58
|
|
|
59
59
|
# Import here to avoid circular imports
|
|
60
|
-
from
|
|
61
|
-
from
|
|
62
|
-
from
|
|
60
|
+
from boltzmann9.data import BMDataset, split_rbm_loaders
|
|
61
|
+
from boltzmann9.model import RBM
|
|
62
|
+
from boltzmann9.utils import resolve_device, resolve_pin_memory
|
|
63
63
|
|
|
64
64
|
# Setup
|
|
65
65
|
device = resolve_device(cfg.get("device", "auto"))
|
|
@@ -127,10 +127,10 @@ def cmd_train(args):
|
|
|
127
127
|
|
|
128
128
|
def cmd_evaluate(args):
|
|
129
129
|
"""Evaluate a trained model."""
|
|
130
|
-
from
|
|
131
|
-
from
|
|
132
|
-
from
|
|
133
|
-
from
|
|
130
|
+
from boltzmann9.config import load_config
|
|
131
|
+
from boltzmann9.pipeline import load_model
|
|
132
|
+
from boltzmann9.project import get_run_paths, find_latest_run
|
|
133
|
+
from boltzmann9.run_utils import save_metrics
|
|
134
134
|
|
|
135
135
|
# Determine run directory
|
|
136
136
|
run_path = Path(args.run)
|
|
@@ -170,9 +170,9 @@ def cmd_evaluate(args):
|
|
|
170
170
|
print("=" * 60)
|
|
171
171
|
|
|
172
172
|
# Import and setup
|
|
173
|
-
from
|
|
174
|
-
from
|
|
175
|
-
from
|
|
173
|
+
from boltzmann9.data import BMDataset, split_rbm_loaders
|
|
174
|
+
from boltzmann9.tester import RBMTester
|
|
175
|
+
from boltzmann9.utils import resolve_device, resolve_pin_memory
|
|
176
176
|
|
|
177
177
|
device = resolve_device(cfg.get("device", "auto"))
|
|
178
178
|
model = model.to(device)
|
|
@@ -242,8 +242,8 @@ def cmd_evaluate(args):
|
|
|
242
242
|
|
|
243
243
|
def cmd_preprocess_raw(args):
|
|
244
244
|
"""Preprocess raw data based on config."""
|
|
245
|
-
from
|
|
246
|
-
from
|
|
245
|
+
from boltzmann9.config import load_config
|
|
246
|
+
from boltzmann9.preprocessor import DataPreprocessor
|
|
247
247
|
|
|
248
248
|
config_path = Path(args.config)
|
|
249
249
|
|
|
@@ -270,7 +270,7 @@ def cmd_preprocess_raw(args):
|
|
|
270
270
|
|
|
271
271
|
def cmd_list_runs(args):
|
|
272
272
|
"""List all runs in a project."""
|
|
273
|
-
from
|
|
273
|
+
from boltzmann9.project import list_runs
|
|
274
274
|
|
|
275
275
|
project_path = Path(args.project)
|
|
276
276
|
|
boltzmann9/model.py
CHANGED
|
@@ -137,7 +137,7 @@ class RBM(nn.Module):
|
|
|
137
137
|
raise FileNotFoundError(f"Config file not found: {config_path}")
|
|
138
138
|
|
|
139
139
|
# Load config
|
|
140
|
-
from
|
|
140
|
+
from boltzmann9.config import load_config
|
|
141
141
|
config = load_config(config_path)
|
|
142
142
|
|
|
143
143
|
# Load checkpoint
|
|
@@ -1,10 +1,10 @@
|
|
|
1
1
|
boltzmann9/__init__.py,sha256=ziEu9BCO40bLdp0HWpLTD32iwpNQSEBsfltOlz0R87A,960
|
|
2
2
|
boltzmann9/__main__.py,sha256=yj4HraCGopPtcUNTFtgTJ9VaXpVHzUWsnqfMOG04Sas,71
|
|
3
|
-
boltzmann9/cli.py,sha256=
|
|
3
|
+
boltzmann9/cli.py,sha256=mPD6QonSOZ3P3Vc5hznK3Ib3Lf_3ROvLBmcY385BFho,11388
|
|
4
4
|
boltzmann9/config.py,sha256=tNvK6HsToX21Gu3KPCgt_6rOi9yYDpRG53zkniSZ2Wc,1604
|
|
5
5
|
boltzmann9/data.py,sha256=xgaHu6DPaanHrptl8ECq31t6E5yD9MatdVb8JHG-l0U,4011
|
|
6
6
|
boltzmann9/data_generator.py,sha256=aXm8b6O4qoXpDa8TCOdDxF_lDaMtcLNR0P06u2cXb1g,7550
|
|
7
|
-
boltzmann9/model.py,sha256=
|
|
7
|
+
boltzmann9/model.py,sha256=npBDd_pow4f9gcsLIpWMmJLBB6sGljtaOluvXcFlLP0,31538
|
|
8
8
|
boltzmann9/pipeline.py,sha256=C3L7tHVakrmvdnW9KYMXhXvscwG7TP8xmVHbunx1E8w,5965
|
|
9
9
|
boltzmann9/preprocessor.py,sha256=USxpMCnC80wp4zO72lPxmFeVdfB4I5NE4P2YlojCVPQ,21758
|
|
10
10
|
boltzmann9/project.py,sha256=Iap-nxhhAZDw2Y4yQg53gFQ0CQPCk2zfmZZC0bTPmxQ,5333
|
|
@@ -12,8 +12,8 @@ boltzmann9/run_utils.py,sha256=1eRDEkN8j1mG5g9iXArWRA1yQq6pw2agVUGV4Yhy0jk,7823
|
|
|
12
12
|
boltzmann9/tester.py,sha256=yMYhYKWWiG-LhJQvKv7QmMCYPQLbmtMxFVrJ9iHZVs8,5659
|
|
13
13
|
boltzmann9/utils.py,sha256=8ftNcbV4dz52-t0Ewu4E7CmkkVzPDa6kd8ZnxaBvhMM,1337
|
|
14
14
|
boltzmann9/visualization.py,sha256=iPSytX7GMfmbRbR43xOmybLZBLmy58D4Rp8wasISMcs,3407
|
|
15
|
-
boltzmann9-0.1.
|
|
16
|
-
boltzmann9-0.1.
|
|
17
|
-
boltzmann9-0.1.
|
|
18
|
-
boltzmann9-0.1.
|
|
19
|
-
boltzmann9-0.1.
|
|
15
|
+
boltzmann9-0.1.6.dist-info/METADATA,sha256=XFCzyIdHSYqWfHM983Mq46xls7M58Kn4q5m7d4WyDyo,3097
|
|
16
|
+
boltzmann9-0.1.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
|
|
17
|
+
boltzmann9-0.1.6.dist-info/entry_points.txt,sha256=Y2wY1lqICCjTxuaPgme2QxkoAknrASKmhLljcb5FEyw,51
|
|
18
|
+
boltzmann9-0.1.6.dist-info/top_level.txt,sha256=AE0_urGKOFSgKxXf7WQfjGDROWYppyI4tw1VIg-paQI,11
|
|
19
|
+
boltzmann9-0.1.6.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|