het_net¶
- probly.method.het_net(base: Predictor[In, Out], num_factors: int = 10, temperature: float = 1.0, is_parameter_efficient: bool = False) HetNetPredictor[In, Out][source]¶
Create a HetNets predictor from a base predictor base on [CMK+21].
- Parameters:
base – The base model to be transformed.
num_factors – The rank of the low-rank covariance parametrization.
temperature – The temperature parameter for scaling the utility.
is_parameter_efficient – Whether to use the parameter-efficient version.
- Returns:
The HetNets predictor.
