def _test_prepare_inputs_for_rnn(self, sequence_features, context_features, num_unroll, batch_size, expected): features_by_time = ssre._prepare_inputs_for_rnn( sequence_features, context_features, num_unroll) with self.test_session() as sess: sess.run(variables.global_variables_initializer()) sess.run(data_flow_ops.initialize_all_tables()) features_val = sess.run(features_by_time) self.assertAllEqual(expected, features_val)
def _test_prepare_inputs_for_rnn(self, sequence_features, context_features, sequence_feature_columns, num_unroll, expected): features_by_time = ssre._prepare_inputs_for_rnn( sequence_features, context_features, sequence_feature_columns, num_unroll) with self.cached_session() as sess: sess.run(variables.global_variables_initializer()) sess.run(lookup_ops.tables_initializer()) features_val = sess.run(features_by_time) self.assertAllEqual(expected, features_val)
def _test_prepare_inputs_for_rnn(self, sequence_features, context_features, sequence_feature_columns, num_unroll, expected): features_by_time = ssre._prepare_inputs_for_rnn(sequence_features, context_features, sequence_feature_columns, num_unroll) with self.test_session() as sess: sess.run(variables.global_variables_initializer()) sess.run(data_flow_ops.initialize_all_tables()) features_val = sess.run(features_by_time) self.assertAllEqual(expected, features_val)