Использование супер в пользовательском обратном вызове Keras - PullRequest
Новая
       31

Использование супер в пользовательском обратном вызове Keras

0 голосов
/ 13 января 2019

Я не очень знаком с функциональностью и наследованием Python super. Я попытался скопировать и использовать пример пользовательского обратного вызова keras, который нашел в этом посте , но я получаю сообщение об ошибке: 

    super(EarlyStopping, self).__init__()
TypeError: super(type, obj): obj must be an instance or subtype of type

Вот пример кода: 

import numpy as np
from tensorflow.keras.callbacks import Callback, EarlyStopping

class OverfitEarlyStopping(Callback):
    def __init__(self, ratio=0.0,
                 patience=0, verbose=0):
        super(EarlyStopping, self).__init__()

        self.ratio = ratio
        self.patience = patience
        self.verbose = verbose
        self.wait = 0
        self.stopped_epoch = 0
        self.monitor_op = np.greater

    def on_train_begin(self, logs=None):
        self.wait = 0  # Allow instances to be re-used

    def on_epoch_end(self, epoch, logs=None):
        current_val = logs.get('val_loss')
        current_train = logs.get('loss')
        if current_val is None:
            warnings.warn('Early stopping requires %s available!' %
                          (self.monitor), RuntimeWarning)

        # If ratio current_loss / current_val_loss > self.ratio
        if self.monitor_op(np.divide(current_train,current_val),self.ratio):
            self.wait = 0
        else:
            if self.wait >= self.patience:
                self.stopped_epoch = epoch
                self.model.stop_training = True
            self.wait += 1

    def on_train_end(self, logs=None):
        if self.stopped_epoch > 0 and self.verbose > 0:
            print('Epoch %05d: early stopping due to overfitting.' % (self.stopped_epoch))

overfit_callback = OverfitEarlyStopping(ratio=0.8, patience=3, verbose=1)

Я использую Python 3.5 и tenorflow.keras. Изменилось ли использование super в версиях, которые я использую, или этот обратный вызов изначально был написан неправильно?

1 Ответ

0 голосов
/ 13 января 2019

Нет необходимости инициализировать super при расширении базового класса keras.callbacks.Callback. Модель передается функциям, которые вы выбираете для переопределения. Вы можете увидеть некоторые примеры здесь

Кроме того, почему бы не использовать tf.keras.callbacks.EarlyStopping

...