Machine learning electron correlation in a disordered medium