5619b6770675a7fedd85cbc5ab19773a3ba94e13,autokeras/net_module.py,NetworkModule,predict,#NetworkModule#Any#,117

Before Change

        model = self.best_model.produce_model()
        model.eval()

        outputs = []
        with torch.no_grad():
            for _, inputs in enumerate(test_loader):
                outputs.append(model(inputs).numpy())
        output = reduce(lambda x, y: np.concatenate((x, y)), outputs)
        return output

    def evaluate(self, test_data):
        Evaluate the performance of the best architecture in terms of the loss.

After Change

        model = self.best_model.produce_model()
        model.eval()

        return Backend.predict(model, test_loader)


class CnnModule(NetworkModule):
     Class to create a CNN module.

Instances