ML-based Physical Layer Key Generation