Noise mitigation in gravitational-wave data with machine learning