Esempio n. 1
0
    def testReader1(self):
        # Create DataReader instance
        data_reader = DataReader(self.filename, self.batch_size,
                                 self.seq_length)

        print(data_reader.vocab)
        print(data_reader.get_tensor(self.filename))
Esempio n. 2
0
    def testReader2(self):
        # Create DataReader instance
        data_reader = DataReader(self.filename, self.batch_size,
                                 self.seq_length)

        tensor = data_reader.get_tensor(self.filename)

        data_reader.generate_batches(tensor)
        x = data_reader.x_batches
        y = data_reader.y_batches

        print(x)
        print(y)