Certain aspects of the present disclosure provide techniques and apparatus for training and using machine learning models to estimate a layout of a spatial area. An example method generally includes estimating a representation of a channel using a machine learning model trained to generate the estimated representation of the channel based on a location of a transmitter in a spatial environment, a location of a receiver in the spatial environment, and a three-dimensional representation of the spatial environment. One or more actions are taken based on the estimated representation of the channel.