Source code for src.gridmind.utils.nn_util

[docs]class NeuralNetworkToTableWrapper:
    def __init__(self, network):
[docs]        self.network = network

[docs]    def __getitem__(self, key):
        return self.network(key).cpu().detach().numpy()

[docs]    def get_network(self):
        return self.network

[docs]    def __call__(self, key):
        return self.network(key).cpu().detach().numpy()

[docs]    def __setitem__(self, key, value):
        raise Exception("Cannot set values in a neural network.")