hookeai.model_architectures.procedures.model_data_scaling.get_fitted_data_scaler

get_fitted_data_scaler(model, features_type)[source]

Get fitted model data scalers.

Parameters:

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

  • features_type (str) –

    Features for which data scaler is required:

    ’features_in’ : Input features

    ’features_out’ : Output features

Returns:

data_scaler – Data scaler.

Return type:

{TorchStandardScaler, TorchMinMaxScaler}