Simple non-linear neural network classifier impl with torch