gluonts.shell.train module

gluonts.shell.train.log_metric(metric: str, value: Any) → None[source]

Emits a log message with a value for a specific metric.

Parameters
  • metric – The name of the metric to be reported.

  • value – The metric value to be reported.

gluonts.shell.train.run_test(env: gluonts.shell.sagemaker.TrainEnv, predictor: gluonts.model.predictor.Predictor, test_dataset: Iterable[Dict[str, Any]]) → None[source]
gluonts.shell.train.run_train(forecaster: gluonts.model.estimator.Estimator, train_dataset: Iterable[Dict[str, Any]], validation_dataset: Optional[Iterable[Dict[str, Any]]]) → gluonts.model.predictor.Predictor[source]
gluonts.shell.train.run_train_and_test(env: gluonts.shell.sagemaker.TrainEnv, forecaster_type: Type[Union[gluonts.model.estimator.Estimator, gluonts.model.predictor.Predictor]]) → None[source]