Basic Neural Network Python Module Definition Computer Science