예제 #1
0
파일: base.py 프로젝트: skn123/pl-cnn
    def get_epoch_test(self):

        assert self.on_memory, "only for data loaded on memory"

        for (batch, idx) in indices_generator(shuffle=False,
                                              batch_size=Cfg.batch_size,
                                              n=self.n_test):
            yield self._X_test[batch], self._y_test[batch], idx
예제 #2
0
    def get_epoch_test(self):

        for (batch, idx) in indices_generator(shuffle=False,
                                              batch_size=Cfg.batch_size,
                                              n=self.n_test):
            yield self._X_test[batch], self._y_test[batch], idx