def split_window(self, features):
    inputs = features[:, self.input_slice, :]
    labels = features[:, self.labels_slice, :]
    if (self.label_columns is not None):
        labels = tf.stack([labels[:, :, self.column_indices[name]] for name in self.label_columns], axis=(- 1))
    inputs.set_shape([None, self.input_width, None])
    labels.set_shape([None, self.label_width, None])
    return (inputs, labels)
def split_window(self, features):
    inputs = features[:, self.input_slice, :]
    labels = features[:, self.labels_slice, :]
    if (self.label_columns is not None):
        labels = tf.stack([labels[:, :, self.column_indices[name]] for name in self.label_columns], axis=(- 1))
    inputs.set_shape([None, self.input_width, None])
    labels.set_shape([None, self.label_width, None])
    return (inputs, labels)