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()