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 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 boltzmann.project import create_project
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 boltzmann.config import load_config
24
- from boltzmann.pipeline import train_rbm, save_model
25
- from boltzmann.project import create_run, save_run_config, get_run_paths
26
- from boltzmann.run_utils import RunLogger, save_history, save_plots
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 boltzmann.data import BMDataset, split_rbm_loaders
61
- from boltzmann.model import RBM
62
- from boltzmann.utils import resolve_device, resolve_pin_memory
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 boltzmann.config import load_config
131
- from boltzmann.pipeline import load_model
132
- from boltzmann.project import get_run_paths, find_latest_run
133
- from boltzmann.run_utils import save_metrics
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 boltzmann.data import BMDataset, split_rbm_loaders
174
- from boltzmann.tester import RBMTester
175
- from boltzmann.utils import resolve_device, resolve_pin_memory
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 boltzmann.config import load_config
246
- from boltzmann.preprocessor import DataPreprocessor
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 boltzmann.project import list_runs
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 boltzmann.config import load_config
140
+ from boltzmann9.config import load_config
141
141
  config = load_config(config_path)
142
142
 
143
143
  # Load checkpoint
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: boltzmann9
3
- Version: 0.1.3
3
+ Version: 0.1.6
4
4
  Summary: Restricted Boltzmann Machine implementation in PyTorch
5
5
  License: MIT
6
6
  Requires-Python: >=3.10
@@ -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=owYKApxpuYfDIb9r6i6vMK672invyjhrA22oTpk_QGg,11370
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=E7EoVyqhfwsv4hlJIjB_2XXUUR_iFMsr_06STL8iieg,31537
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.3.dist-info/METADATA,sha256=45XJzNlKbNB2lTQmkEhzHYY3RSVCYyIx9VZ0zsfFaL8,3097
16
- boltzmann9-0.1.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
17
- boltzmann9-0.1.3.dist-info/entry_points.txt,sha256=Y2wY1lqICCjTxuaPgme2QxkoAknrASKmhLljcb5FEyw,51
18
- boltzmann9-0.1.3.dist-info/top_level.txt,sha256=AE0_urGKOFSgKxXf7WQfjGDROWYppyI4tw1VIg-paQI,11
19
- boltzmann9-0.1.3.dist-info/RECORD,,
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,,