AI for Neural Network Class