Basic Neural Network Python Code For Rock