Beispiel #1
0
    def load(cls, dir):
        with open(dir, 'r') as f:
            data = json.load(f)
        layers = []
        for layer in data['layers']:
            if len(layers) == 0:
                layers.append(Layer.from_dict(layer, data['input_nodes']))
            else:
                layers.append(
                    Layer.from_dict(layer, data['layers'][-1]['num_nodes']))

        model = cls(input_nodes=data['input_nodes'],
                    layers=layers,
                    lr=data['lr'],
                    epochs=data['epochs'])
        return model