Python temporary_train_env Examples

Programming Language: Python

Namespace/Package Name: gluonts.testutil.shell

Method/Function: temporary_train_env

Examples at hotexamples.com: 2

Python temporary_train_env - 2 examples found. These are the top rated real world Python examples of gluonts.testutil.shell.temporary_train_env extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

def train_env() -> ContextManager[TrainEnv]:
    hyperparameters = {
        "context_length": context_length,
        "prediction_length": prediction_length,
        "num_samples": num_samples,
    }
    with testutil.temporary_train_env(hyperparameters, "constant") as env:
        yield env

Example #2

Show file

def train_env(listify_dataset) -> ContextManager[TrainEnv]:
    hyperparameters = {
        "context_length": context_length,
        "prediction_length": prediction_length,
        "num_samples": num_samples,
        "listify_dataset": listify_dataset,
        "num_workers": 3,
        "num_prefetch": 4,
        "shuffle_buffer_length": 256,
        "epochs": 3,
    }
    with testutil.temporary_train_env(hyperparameters, "constant") as env:
        yield env