예제 #1
0
def test_time_series_input_node_deserialize_build_no_error():
    node = nodes.TimeseriesInput(lookback=2, shape=(32, ))
    node = nodes.deserialize(nodes.serialize(node))
    hp = keras_tuner.HyperParameters()

    input_node = node.build_node(hp)
    node.build(hp, input_node)
예제 #2
0
def test_time_series_input_node_build_to_a_tensor():
    node = nodes.TimeseriesInput(lookback=2, shape=(32, ))
    hp = kerastuner.HyperParameters()

    input_node = node.build_node(hp)
    output = node.build(hp, input_node)

    assert isinstance(output, tf.Tensor)
예제 #3
0
 def __init__(self,
              outputs,
              column_names=None,
              column_types=None,
              lookback=None,
              predict_from=1,
              predict_until=None,
              **kwargs):
     inputs = input_module.TimeseriesInput(lookback=lookback,
                                           column_names=column_names,
                                           column_types=column_types)
     super().__init__(inputs=inputs, outputs=outputs, **kwargs)
     self.predict_from = predict_from
     self.predict_until = predict_until
     self._target_col_name = None
     self.train_len = 0