def _test_confidence_intervals(dataset, group_compose=False): pipeline = DeepARPipeline(epochs=1, steps_per_epoch=1, number_samples=10, prediction_length=min_pred_lengths[dataset][0], context_length=min_pred_lengths[dataset][0], group_compose=group_compose, confidence_intervals=True) pipeline.write_pipeline() pipeline.fit_produce(dataset) pipeline.delete_pipeline()
def _test_serialize(dataset, group_compose=False): pipeline = DeepARPipeline( epochs=1, steps_per_epoch=1, number_samples=10, prediction_length=min_pred_lengths[dataset][0], context_length=min_pred_lengths[dataset][0], group_compose=group_compose, ) pipeline.write_pipeline() pipeline.fit_serialize(dataset) pipeline.deserialize_score(dataset) pipeline.delete_pipeline() pipeline.delete_serialized_pipeline()