Basic Neural Network Python Module Vs Function