A Neural Network Approach To Treatment Optimization