Reconstruction of electron radiation belts using data assimilation and machine learning