jacobian

a basic keras-like neural network library for c++/python
Log | Files | Refs | README

example.py (587B)


      1 import jacobian as jcb
      2 from matplotlib.animation import FuncAnimation
      3 import matplotlib.pyplot as plt
      4 import numpy as np
      5 import time
      6 net = jcb.Network("./data_banknote_authentication.txt", 10, 0.0155, 0.03, jcb.L2, 1, 0.9)
      7 net.add_layer(4, jcb.activations.linear, jcb.activations.linear_deriv)
      8 net.add_layer(5, jcb.activations.sigmoid, jcb.activations.sigmoid_deriv)
      9 net.add_layer(2, jcb.activations.linear, jcb.activations.linear_deriv)
     10 net.init_optimizer(jcb.optimizers.momentum(0.1))
     11 net.init_decay(jcb.decays.exponential(1, 0.5))
     12 net.initialize()
     13 for i in range(50):
     14     net.train()