The code customizes the example from here: https://github.com/HIPS/autograd/blob/master/examples/neural_net.py
In the kafnets module you can find the code to initialize and run neural networks having KAF activation functions. Note that this is a regression example and we use custom functions also in the output layer. For classification, consider changing the final layer to a standard softmax.
- autograd = 1.2.
- scikit-learn = 0.20.1 (for demo_kaf_regression.py)