tf.contrib.timeseries.OneShotPredictionHead

Class OneShotPredictionHead

Defined in tensorflow/contrib/timeseries/python/timeseries/head.py.

A time series head which exports a single stateless serving signature.

The serving default signature exported by this head expects times, values, and any exogenous features, but no state. values has shape [batch_size, filter_length, num_features] and times has shape [batch_size, total_length], where total_length > filter_length. Any exogenous features must have their shapes prefixed by the shape of the times feature.

When serving, first performs filtering on the series up to filter_length starting from the default start state for the model, then computes predictions on the remainder of the series, returning them.

Model state is neither accepted nor returned, so filtering must be performed each time predictions are requested when using this head.

__init__

__init__(
    model,
    state_manager,
    optimizer,
    input_statistics_generator=None,
    name=None
)

Creates a _Head for time series regression.

Args:

  • model: A model for time series regression.
  • state_manager: A state manager.
  • optimizer: An optimizer.
  • input_statistics_generator: A input statistics generator.
  • name: An optional name for the model.

Properties

logits_dimension

See _Head.

name

The name of this head.

Returns:

A string.

Methods

create_estimator_spec

create_estimator_spec(
    features,
    mode,
    labels=None
)

Performs basic error checking and returns an EstimatorSpec.

create_loss

create_loss(
    features,
    mode,
    logits=None,
    labels=None
)

See _Head.