Solving Train Maintenance Scheduling Problem with Neural Networks and Tree Search