Problem when using [email protected]

[Kiris-tingna]

advise

we need to calc steps_per_epoch and specific this value

    def fit(self, log=False, **kwargs):
        """kwargs: keyword arguments passed to `model.fit`"""
        if log:
            callbacks = [keras.callbacks.TensorBoard(
                log_dir='./log', histogram_freq=0,
                write_graph=True, write_images=False)]
        else:
            callbacks = []

        callbacks += kwargs.get('callbacks', [])
        if 'callbacks' in kwargs:
            del kwargs['callbacks']

        if 'batch_size' in kwargs:
            batch_size = kwargs['batch_size']
            del kwargs['batch_size']
            gen = self.train_data_generator(batch_size=batch_size)
        else:
            raise ValueError("Must Specify batch_size")
            gen = self.train_data_generator()

        m = self.graph.number_of_edges()
        steps_per_epoch = math.ceil(m / batch_size)

        self.model.fit_generator(
            gen,
            steps_per_epoch=steps_per_epoch,
            shuffle=True,
            callbacks=callbacks,
            **kwargs)