hookeai.model_architectures.procedures.model_data_scaling.set_data_scalers

set_data_scalers(model, scaler_features_in, scaler_features_out)[source]

Set fitted model data scalers.

Parameters:

  • model (torch.nn.Module) – Model.

  • scaler_features_in ({TorchStandardScaler, TorchMinMaxScaler}) – Data scaler for input features.

  • scaler_features_out ({TorchStandardScaler, TorchMinMaxScaler}) – Data scaler for output features.