vx-one / models /__init__.py
jsantillana's picture
Add VX-One source code
908fd1a verified
Raw History Blame Contribute Delete
1.06 kB
"""Motor VX-One (spec F2): red multi-cabeza, losses, datos, calibración,
loop de entrenamiento y contrato de inferencia System One."""
from .calibration import (fit_choice_temperature, fit_corn_temperature,
fit_noul_temperature, fit_temperature_scalar)
from .data import EmberBatcher, RunData, build_run
from .engine import VXOneEngine
from .feature_stage import FeatureStage
from .losses import (KendallMultiTask, choice_loss, class_weights, corn_loss,
corn_pmf, noul_loss)
from .network import SCALES, HeadSpec, VXOneNet, head_spec_for
from .train import TRAIN_TABLE, train_run
from .vocab import TextVocab
__all__ = [
"VXOneNet", "HeadSpec", "head_spec_for", "SCALES",
"FeatureStage", "TextVocab",
"class_weights", "noul_loss", "choice_loss", "corn_loss", "corn_pmf",
"KendallMultiTask",
"fit_temperature_scalar", "fit_noul_temperature", "fit_choice_temperature",
"fit_corn_temperature",
"RunData", "build_run", "EmberBatcher",
"VXOneEngine", "train_run", "TRAIN_TABLE",
]