88d3f856936b8620e999386476bcdd0f96ae846b,model/uisrnn.py,UISRNN,fit,#UISRNN#,151

Before Change

                       "train_cluster_id length.")
    if type(train_sequence).__module__ != np.__name__:
      raise TypeError("train_sequence type should be an numpy array.")
    if type(train_cluster_id).__module__ != np.__name__:
      raise TypeError("train_cluster_id type should be an numpy array.")

    self.rnn_model.train()

After Change

    // check dimension
    if train_sequence.ndim != 2:
      raise ValueError("train_sequence must be 2-dim array")
    if train_cluster_id.ndim != 1:
      raise ValueError("train_cluster_id must be 1-dim array")
    // check length and size
    train_total_length, observation_dim = train_sequence.shape
    if observation_dim != self.observation_dim:
      raise ValueError("train_sequence does not match the dimension specified "
                       "by args.observation_dim.")

Instances