BERNN-MSMS
Minimal README for quick usage.
Longer historical content is kept in LEGACY_README.md.
Install
pip install bernn
Basic usage
from bernn import TrainAEClassifierHoldout
trainer_cls = TrainAEClassifierHoldout
trainer = trainer_cls(config=bernn_config, log_metrics=True, keep_models=False)
# Train and predict in one call
preds_encoded = trainer.fit_predict(
X_train,
y_train,
X_test=X_test,
y_test=y_test,
groups_train=batches_train,
groups_test=batches_test,
cross_validation=False,
cross_test=False,
)
# Decode predictions back to original labels
preds = trainer.predict(X_test)
Important runtime contract:
- groups_train is mandatory.
- If X_test is provided, groups_test is mandatory.
Important parameters
Focus on these first:
- optimize_hyperparams: enable/disable Ax optimization.
- n_trials: number of optimization trials.
- fixed_hyperparams: force values and remove them from search.
- n_repeats: number of holdout repeats.
- n_layers, layer1: classifier depth and width seed.
- dloss: domain loss mode.
- warmup, n_epochs: core training schedule.
- device: cpu/cuda target.
- scaler, bs: preprocessing and batch size.
Official documentation
- Full reference: OFFICIAL_DOCUMENTATION.md
- Full parameter catalog: TRAINING_PARAMETERS.md
- Minimal runnable examples notebook (4 variants): tutorials/minimal_examples.ipynb
- Optimized all-config notebook (TrainAEClassifierHoldout): tutorials/optimized_classifier_holdout_all_configs.ipynb
- Optimized all-config notebook (TrainAEThenClassifierHoldout): tutorials/optimized_ae_then_classifier_holdout_all_configs.ipynb
- Historical CLI-heavy guide: LEGACY_README.md
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
bernn-0.6.15.tar.gz
(293.0 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
bernn-0.6.15-py3-none-any.whl
(312.9 kB
view details)
File details
Details for the file bernn-0.6.15.tar.gz.
File metadata
- Download URL: bernn-0.6.15.tar.gz
- Upload date:
- Size: 293.0 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.10.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
ac2d46f43dd15db778addc93a7620655e669b95d5f1fef880f2bd0f2082d734f
|
|
| MD5 |
e2cb4fcbbe76bc4c15e583a5cb1d4aef
|
|
| BLAKE2b-256 |
744e973e6c5a7047ffe373dc78f3e8d6c7199308575032a3582fee3998d47cd1
|
File details
Details for the file bernn-0.6.15-py3-none-any.whl.
File metadata
- Download URL: bernn-0.6.15-py3-none-any.whl
- Upload date:
- Size: 312.9 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.10.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
391d08a776c37f3bb332b726f2b7a696b7f72f6509462469ebbbc86b6d9ec927
|
|
| MD5 |
d6f0391a1e02749c4db83e20098cfd9a
|
|
| BLAKE2b-256 |
4da11a37330aa4e2bf209cc3c948118507b2c612f72505a3c9cc7c7e5e71a225
|
