saberlib 0.1.0__tar.gz

saberlib-0.1.0/.gitignore

@@ -0,0 +1,26 @@
+# Python
+__pycache__/
+*.py[cod]
+
+# Build
+build/
+dist/
+*.egg-info/
+
+# Environments
+.venv/
+.env
+
+# Tooling caches and reports
+.pytest_cache/
+.ruff_cache/
+.coverage
+.coverage.*
+htmlcov/
+
+# Marimo
+__marimo__/
+
+# Editors
+.idea/
+.vscode/

saberlib-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Kren-AI Lab
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

saberlib-0.1.0/PKG-INFO

@@ -0,0 +1,209 @@
+Metadata-Version: 2.5
+Name: saberlib
+Version: 0.1.0
+Summary: Domain-agnostic classical supervised machine learning for classification, regression, tuning, and benchmarking.
+Project-URL: Homepage, https://github.com/kren-ai-lab/saber
+Project-URL: Repository, https://github.com/kren-ai-lab/saber
+Project-URL: Issues, https://github.com/kren-ai-lab/saber/issues
+Project-URL: Documentation, https://github.com/kren-ai-lab/saber#readme
+Author: Diego Alvarez-Saravia, David Medina-Ortiz
+Maintainer: Diego Alvarez-Saravia, David Medina-Ortiz
+License-Expression: MIT
+License-File: LICENSE
+Keywords: benchmarking,classification,hyperparameter-optimization,machine-learning,regression,reproducibility,scientific-software,supervised-learning,tabular-data
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Typing :: Typed
+Requires-Python: <3.15,>=3.11
+Requires-Dist: joblib<2,>=1.3
+Requires-Dist: numpy<3,>=2.0
+Requires-Dist: polars<2,>=1.44
+Requires-Dist: pyyaml<7,>=6.0
+Requires-Dist: rich<16,>=13.0
+Requires-Dist: scikit-learn<2,>=1.8
+Requires-Dist: scipy<2,>=1.16
+Requires-Dist: typer<1,>=0.27
+Provides-Extra: all
+Requires-Dist: biosieve<0.2,>=0.1.2; extra == 'all'
+Requires-Dist: lightgbm<5,>=4.6; extra == 'all'
+Requires-Dist: optuna<5,>=4.6; extra == 'all'
+Requires-Dist: xgboost<4,>=2.0; extra == 'all'
+Provides-Extra: biosieve
+Requires-Dist: biosieve<0.2,>=0.1.2; extra == 'biosieve'
+Provides-Extra: lightgbm
+Requires-Dist: lightgbm<5,>=4.6; extra == 'lightgbm'
+Provides-Extra: optuna
+Requires-Dist: optuna<5,>=4.6; extra == 'optuna'
+Provides-Extra: xgboost
+Requires-Dist: xgboost<4,>=2.0; extra == 'xgboost'
+Description-Content-Type: text/markdown
+
+# Saber
+
+[![PyPI](https://img.shields.io/pypi/v/saberlib?style=flat-square)](https://pypi.org/project/saberlib/)
+[![PyVersions](https://img.shields.io/pypi/pyversions/saberlib?style=flat-square)](https://github.com/kren-ai-lab/saber)
+[![Tests](https://img.shields.io/github/actions/workflow/status/kren-ai-lab/saber/tests.yml?style=flat-square)](https://github.com/kren-ai-lab/saber/actions/workflows/tests.yml)
+![License](https://img.shields.io/github/license/kren-ai-lab/saber?style=flat-square)
+
+Saber is a Python library for classical supervised machine learning
+(classification and regression) on numerical tabular features. It trains,
+validates, tunes, benchmarks and persists scikit-learn, XGBoost and LightGBM
+models, with preprocessing fitted inside each fold and every result tied back
+to its samples, partition and parameters.
+
+Saber doesn't compute features. Bring a descriptor table, an embedding or any
+numeric matrix, and Saber models it. Deep learning, AutoML, multilabel and
+multi-output problems are out of scope.
+
+## Installation
+
+Saber supports Python 3.11 to 3.14.
+
+```bash
+python -m pip install saberlib
+```
+
+Optional extras:
+
+```bash
+python -m pip install "saberlib[biosieve]"   # partition generation
+python -m pip install "saberlib[optuna]"     # Optuna tuning
+python -m pip install "saberlib[xgboost]"    # XGBoost models
+python -m pip install "saberlib[lightgbm]"   # LightGBM models
+python -m pip install "saberlib[all]"
+```
+
+For development setup with
+`uv`, see [DEVELOPMENT.md](DEVELOPMENT.md).
+
+## Validate a model
+
+```python
+import numpy as np
+from sklearn.datasets import make_classification
+
+import saber
+from saber import DatasetBundle, PartitionPlan
+
+X, y = make_classification(n_samples=120, n_features=12, random_state=42)
+dataset = DatasetBundle(X=X, y=y, sample_ids=[f"s{i}" for i in range(len(y))])
+
+plan = PartitionPlan.from_predefined_folds(
+    sample_ids=dataset.sample_ids,
+    fold_assignments=np.arange(len(y)) % 4,
+    dataset=dataset,
+)
+
+result = saber.validate(
+    dataset=dataset,
+    algorithm="logistic_regression",
+    partition_plan=plan,
+    metrics=("mcc", "balanced_accuracy", "roc_auc"),
+    random_state=42,
+)
+print(result.aggregate_metrics)
+```
+
+`result.oof_prediction` holds the out-of-fold predictions, aligned by sample
+ID. `X` can be a NumPy array, a Polars DataFrame or a pandas DataFrame. Partitions
+refer to samples by ID, never by row position. If your data isn't split yet,
+let BioSieve generate the partitions:
+
+```python
+from saber import BioSievePartitionConfig
+
+result = saber.validate(
+    dataset=dataset,
+    algorithm="random_forest_classifier",
+    partition_plan=BioSievePartitionConfig(
+        strategy="stratified_kfold",
+        params={"n_splits": 5, "seed": 42},
+    ),
+    metrics=("mcc", "roc_auc"),
+)
+```
+
+The [data and partitions](docs/data_and_partitions.md) guide covers holdouts,
+external partition files and BioSieve.
+
+## Tune, benchmark and save
+
+```python
+from saber import Categorical, LogFloat, SearchSpace, TuningConfig
+
+tuned = saber.tune(
+    dataset=dataset,
+    algorithm="svc",
+    partition_plan=plan,
+    search_space=SearchSpace(
+        parameters={"C": LogFloat(1e-3, 1e2), "kernel": Categorical(["linear", "rbf"])},
+    ),
+    config=TuningConfig(optimizer="random", n_trials=20),
+    metrics=("mcc",),
+    random_state=42,
+)
+print(tuned.best_params)
+```
+
+`saber.benchmark(...)` crosses representations, partitions, algorithms, seeds
+and tuned/untuned modes, and returns long-form Polars tables of metrics and
+per-sample predictions. `saber.train(...)` fits a final model, and
+`saber.save_model(path, result, dataset=dataset)` writes a `train` or `tune` result as a directory with a feature schema,
+provenance and checksums that `saber.load_model(...)` verifies before loading.
+See the [tuning](docs/tuning.md), [benchmarking](docs/benchmarking.md) and
+[persistence](docs/persistence.md) guides.
+
+## Command line
+
+```bash
+saber models list --task classification
+saber run experiment.yaml --dry-run
+saber run study.yaml --json
+saber artifact verify artifacts/model
+```
+
+Every workflow can be written as a YAML or JSON file and run from the CLI or
+with `saber.run_config("experiment.yaml")`. See the [configuration](docs/configuration.md)
+and [CLI](docs/cli.md) references.
+
+## Design principles
+
+- Imputation and scaling are fitted inside each training fold, never on the
+  whole dataset.
+- Existing partitions are used exactly as given. New ones come from BioSieve;
+  Saber has no splitters of its own.
+- Tuned results are reported on a protected test set, not on the folds used to
+  pick the hyperparameters.
+- For binary problems the positive class is explicit, never inferred from a
+  probability column's position.
+- Saved models carry dataset and partition fingerprints and are checksum
+  verified before loading. They use joblib, so load them only from trusted
+  sources.
+
+Saber keeps the workflow honest, but it can't tell whether your partitions or
+metrics answer your scientific question, or catch leakage that happened
+upstream.
+
+## Documentation and examples
+
+The [documentation index](docs/README.md) links all the guides. There are also
+[13 example notebooks](examples/README.md) written in marimo. To open one:
+
+```bash
+uv sync --all-extras --group examples
+uv run marimo edit examples/01_binary_classification.py
+```
+
+## Citing and license
+
+If you use Saber in published work, please cite it using
+[`CITATION.cff`](https://github.com/kren-ai-lab/saber/blob/main/CITATION.cff), or the "Cite this repository" button on
+GitHub. Saber is released under the [MIT license](https://github.com/kren-ai-lab/saber/blob/main/LICENSE).

saberlib-0.1.0/README.md

@@ -0,0 +1,161 @@
+# Saber
+
+[![PyPI](https://img.shields.io/pypi/v/saberlib?style=flat-square)](https://pypi.org/project/saberlib/)
+[![PyVersions](https://img.shields.io/pypi/pyversions/saberlib?style=flat-square)](https://github.com/kren-ai-lab/saber)
+[![Tests](https://img.shields.io/github/actions/workflow/status/kren-ai-lab/saber/tests.yml?style=flat-square)](https://github.com/kren-ai-lab/saber/actions/workflows/tests.yml)
+![License](https://img.shields.io/github/license/kren-ai-lab/saber?style=flat-square)
+
+Saber is a Python library for classical supervised machine learning
+(classification and regression) on numerical tabular features. It trains,
+validates, tunes, benchmarks and persists scikit-learn, XGBoost and LightGBM
+models, with preprocessing fitted inside each fold and every result tied back
+to its samples, partition and parameters.
+
+Saber doesn't compute features. Bring a descriptor table, an embedding or any
+numeric matrix, and Saber models it. Deep learning, AutoML, multilabel and
+multi-output problems are out of scope.
+
+## Installation
+
+Saber supports Python 3.11 to 3.14.
+
+```bash
+python -m pip install saberlib
+```
+
+Optional extras:
+
+```bash
+python -m pip install "saberlib[biosieve]"   # partition generation
+python -m pip install "saberlib[optuna]"     # Optuna tuning
+python -m pip install "saberlib[xgboost]"    # XGBoost models
+python -m pip install "saberlib[lightgbm]"   # LightGBM models
+python -m pip install "saberlib[all]"
+```
+
+For development setup with
+`uv`, see [DEVELOPMENT.md](DEVELOPMENT.md).
+
+## Validate a model
+
+```python
+import numpy as np
+from sklearn.datasets import make_classification
+
+import saber
+from saber import DatasetBundle, PartitionPlan
+
+X, y = make_classification(n_samples=120, n_features=12, random_state=42)
+dataset = DatasetBundle(X=X, y=y, sample_ids=[f"s{i}" for i in range(len(y))])
+
+plan = PartitionPlan.from_predefined_folds(
+    sample_ids=dataset.sample_ids,
+    fold_assignments=np.arange(len(y)) % 4,
+    dataset=dataset,
+)
+
+result = saber.validate(
+    dataset=dataset,
+    algorithm="logistic_regression",
+    partition_plan=plan,
+    metrics=("mcc", "balanced_accuracy", "roc_auc"),
+    random_state=42,
+)
+print(result.aggregate_metrics)
+```
+
+`result.oof_prediction` holds the out-of-fold predictions, aligned by sample
+ID. `X` can be a NumPy array, a Polars DataFrame or a pandas DataFrame. Partitions
+refer to samples by ID, never by row position. If your data isn't split yet,
+let BioSieve generate the partitions:
+
+```python
+from saber import BioSievePartitionConfig
+
+result = saber.validate(
+    dataset=dataset,
+    algorithm="random_forest_classifier",
+    partition_plan=BioSievePartitionConfig(
+        strategy="stratified_kfold",
+        params={"n_splits": 5, "seed": 42},
+    ),
+    metrics=("mcc", "roc_auc"),
+)
+```
+
+The [data and partitions](docs/data_and_partitions.md) guide covers holdouts,
+external partition files and BioSieve.
+
+## Tune, benchmark and save
+
+```python
+from saber import Categorical, LogFloat, SearchSpace, TuningConfig
+
+tuned = saber.tune(
+    dataset=dataset,
+    algorithm="svc",
+    partition_plan=plan,
+    search_space=SearchSpace(
+        parameters={"C": LogFloat(1e-3, 1e2), "kernel": Categorical(["linear", "rbf"])},
+    ),
+    config=TuningConfig(optimizer="random", n_trials=20),
+    metrics=("mcc",),
+    random_state=42,
+)
+print(tuned.best_params)
+```
+
+`saber.benchmark(...)` crosses representations, partitions, algorithms, seeds
+and tuned/untuned modes, and returns long-form Polars tables of metrics and
+per-sample predictions. `saber.train(...)` fits a final model, and
+`saber.save_model(path, result, dataset=dataset)` writes a `train` or `tune` result as a directory with a feature schema,
+provenance and checksums that `saber.load_model(...)` verifies before loading.
+See the [tuning](docs/tuning.md), [benchmarking](docs/benchmarking.md) and
+[persistence](docs/persistence.md) guides.
+
+## Command line
+
+```bash
+saber models list --task classification
+saber run experiment.yaml --dry-run
+saber run study.yaml --json
+saber artifact verify artifacts/model
+```
+
+Every workflow can be written as a YAML or JSON file and run from the CLI or
+with `saber.run_config("experiment.yaml")`. See the [configuration](docs/configuration.md)
+and [CLI](docs/cli.md) references.
+
+## Design principles
+
+- Imputation and scaling are fitted inside each training fold, never on the
+  whole dataset.
+- Existing partitions are used exactly as given. New ones come from BioSieve;
+  Saber has no splitters of its own.
+- Tuned results are reported on a protected test set, not on the folds used to
+  pick the hyperparameters.
+- For binary problems the positive class is explicit, never inferred from a
+  probability column's position.
+- Saved models carry dataset and partition fingerprints and are checksum
+  verified before loading. They use joblib, so load them only from trusted
+  sources.
+
+Saber keeps the workflow honest, but it can't tell whether your partitions or
+metrics answer your scientific question, or catch leakage that happened
+upstream.
+
+## Documentation and examples
+
+The [documentation index](docs/README.md) links all the guides. There are also
+[13 example notebooks](examples/README.md) written in marimo. To open one:
+
+```bash
+uv sync --all-extras --group examples
+uv run marimo edit examples/01_binary_classification.py
+```
+
+## Citing and license
+
+If you use Saber in published work, please cite it using
+[`CITATION.cff`](https://github.com/kren-ai-lab/saber/blob/main/CITATION.cff), or the "Cite this repository" button on
+GitHub. Saber is released under the [MIT license](https://github.com/kren-ai-lab/saber/blob/main/LICENSE).

saberlib-0.1.0/pyproject.toml

@@ -0,0 +1,174 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "saberlib"
+dynamic = ["version"]
+description = "Domain-agnostic classical supervised machine learning for classification, regression, tuning, and benchmarking."
+readme = { file = "README.md", content-type = "text/markdown" }
+requires-python = ">=3.11,<3.15"
+license = "MIT"
+license-files = ["LICENSE"]
+authors = [
+    { name = "Diego Alvarez-Saravia" },
+    { name = "David Medina-Ortiz" },
+]
+maintainers = [
+    { name = "Diego Alvarez-Saravia" },
+    { name = "David Medina-Ortiz" },
+]
+keywords = [
+    "machine-learning",
+    "supervised-learning",
+    "tabular-data",
+    "classification",
+    "regression",
+    "benchmarking",
+    "hyperparameter-optimization",
+    "reproducibility",
+    "scientific-software",
+]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Science/Research",
+    "Intended Audience :: Developers",
+    "Operating System :: OS Independent",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Typing :: Typed",
+]
+
+dependencies = [
+    "numpy>=2.0,<3",
+    "scikit-learn>=1.8,<2",
+    "scipy>=1.16,<2",
+    "joblib>=1.3,<2",
+    "rich>=13.0,<16",
+    "pyyaml>=6.0,<7",
+    "typer>=0.27,<1",
+    "polars>=1.44,<2",
+]
+
+[project.scripts]
+saber = "saber.cli.main:main"
+
+[project.optional-dependencies]
+biosieve = ["biosieve>=0.1.2,<0.2"]
+xgboost = ["xgboost>=2.0,<4"]
+lightgbm = ["lightgbm>=4.6,<5"]
+optuna = ["optuna>=4.6,<5"]
+all = [
+    "biosieve>=0.1.2,<0.2",
+    "xgboost>=2.0,<4",
+    "lightgbm>=4.6,<5",
+    "optuna>=4.6,<5",
+]
+
+[project.urls]
+Homepage = "https://github.com/kren-ai-lab/saber"
+Repository = "https://github.com/kren-ai-lab/saber"
+Issues = "https://github.com/kren-ai-lab/saber/issues"
+Documentation = "https://github.com/kren-ai-lab/saber#readme"
+
+[dependency-groups]
+dev = [
+    "pytest>=9.0",
+    "pytest-cov>=7.1",
+    "ruff>=0.16",
+    "pyrefly>1.0",
+    "taskipy>=1.14",
+    "prek",
+    "pandas>=2.2.2",
+    "pyarrow>=25.0.1",
+]
+examples = [
+    "marimo>=0.24",
+    "matplotlib>=3.8",
+]
+
+[tool.hatch.version]
+path = "saber/_version.py"
+pattern = '__version__ = "(?P<version>[^"]+)"'
+
+[tool.hatch.build.targets.sdist]
+include = ["/saber"]
+
+[tool.hatch.build.targets.wheel]
+packages = ["saber"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+addopts = "-ra"
+
+[tool.ruff]
+line-length = 110
+target-version = "py311"
+
+[tool.ruff.lint]
+select = ["ALL"]
+ignore = [
+    "COM812",
+    "CPY001",
+    "D203",
+    "D213",
+    "N803",
+    "N806",
+    "PLR0913",
+    # Deliberate exceptions, not pending work.
+    "TRY003",  # a raise names the offending data, not a class per failure mode
+    "EM101",
+    "EM102",
+    "ANN401",  # coercion helpers take anything; `object` only moves it to pyrefly
+    "PLR2004", # `if n < 3` reads better than a named constant
+    "C901",    # workflows branch on contracts; splitting them hides the execution path
+    "PLR0911",
+    "PLR0912",
+    "PLR0915",
+    "PLR0917",
+]
+
+[tool.ruff.lint.per-file-ignores]
+"saber/cli/*" = [
+    "FBT001",  # bool flags are standard in CLI functions
+    "FBT002",
+    "FBT003",
+    "TC001",   # typer resolves annotations at runtime, so they stay imported
+    "TC002",
+    "TC003",
+]
+"tests/*" = [
+    "ANN",     # test helpers are not part of the typed public surface
+    "D",       # docstrings are not required in tests
+    "INP001",  # test packages are intentionally implicit namespaces
+    "PLC0415",
+    "PT011",
+    "PT017",
+    "PT018",
+    "S101",    # assert is how a test asserts
+    "TC003",   # pytest resolves fixture annotations at runtime
+]
+"examples/*" = [
+    "INP001",  # example scripts are intentionally standalone modules
+    "T201",    # examples print result snippets by design
+]
+
+[tool.pyrefly]
+# Without a config section Pyrefly falls back to the `basic` preset, which
+# silences most of its checks; `default` is the real baseline.
+preset = "default"
+project-includes = ["saber", "tests"]
+
+[tool.taskipy.tasks]
+sort-imports = "ruff check --fix --select I,F401 saber/ tests/"
+format = "task sort-imports && ruff format saber/ tests/"
+lint = "ruff check saber/ tests/"
+lint-fix = "ruff check --fix saber/ tests/"
+test = "pytest -q"
+test-v = "pytest -v"
+test-cov = "pytest --cov=saber --cov-report=html"
+pyrefly = "pyrefly check saber/ tests/"

saberlib-0.1.0/saber/__init__.py

@@ -0,0 +1,71 @@
+"""saber — classical supervised machine learning infrastructure."""
+
+from saber._api import evaluate, predict, train
+from saber._version import __version__
+from saber.benchmark import BenchmarkConfig, BenchmarkResult, benchmark
+from saber.config import run_config
+from saber.core import (
+    ALGORITHMS,
+    Categorical,
+    Float,
+    Integer,
+    LogFloat,
+    PredictionResult,
+    SearchSpace,
+    TrainResult,
+)
+from saber.datasets import BioSievePartitionConfig, DatasetBundle, PartitionPlan
+from saber.evaluation import EvaluationResult
+from saber.exceptions import SaberError
+from saber.persistence import (
+    LoadedModelArtifact,
+    inspect_artifact,
+    load_benchmark,
+    load_model,
+    save_benchmark,
+    save_model,
+)
+from saber.preprocessing import PreprocessingConfig
+from saber.tuning import OptimizationResult, TuningConfig, tune
+from saber.validation import ValidationResult, validate
+
+__all__ = [  # noqa: RUF022  # grouped by purpose, not alphabetical
+    # Workflows
+    "train",
+    "validate",
+    "tune",
+    "benchmark",
+    "evaluate",
+    "predict",
+    "run_config",
+    # Persistence
+    "save_model",
+    "load_model",
+    "save_benchmark",
+    "load_benchmark",
+    "inspect_artifact",
+    # Inputs
+    "DatasetBundle",
+    "PartitionPlan",
+    "BioSievePartitionConfig",
+    "PreprocessingConfig",
+    "TuningConfig",
+    "BenchmarkConfig",
+    "SearchSpace",
+    "Categorical",
+    "Integer",
+    "Float",
+    "LogFloat",
+    # Results
+    "TrainResult",
+    "PredictionResult",
+    "EvaluationResult",
+    "ValidationResult",
+    "OptimizationResult",
+    "BenchmarkResult",
+    "LoadedModelArtifact",
+    # Catalog / misc
+    "ALGORITHMS",
+    "SaberError",
+    "__version__",
+]

saberlib-0.1.0/saber/__main__.py

@@ -0,0 +1,5 @@
+"""Entry point for running saber's CLI as a module."""
+
+from saber.cli.main import main
+
+raise SystemExit(main())

saberlib-0.1.0/saber/_api/__init__.py

@@ -0,0 +1,7 @@
+"""Private implementation of the train/predict/evaluate workflows."""
+
+from saber._api.evaluate import evaluate
+from saber._api.predict import predict
+from saber._api.train import train
+
+__all__ = ["evaluate", "predict", "train"]