Searched refs:TimeSeriesRegressionHead (Results 1 – 3 of 3) sorted by relevance
51 config=None, head_type=ts_head_lib.TimeSeriesRegressionHead):550 config=None, head_type=ts_head_lib.TimeSeriesRegressionHead):608 head_type=ts_head_lib.TimeSeriesRegressionHead):
51 class TimeSeriesRegressionHead(head_lib._Head): # pylint:disable=protected-access class280 class OneShotPredictionHead(TimeSeriesRegressionHead):
115 model_fn = ts_head_lib.TimeSeriesRegressionHead(191 return ts_head_lib.TimeSeriesRegressionHead(