Text this: Neural Network-Based Imitation Learning for Approximating Stochastic Battery Management Systems