Learn R Programming

shapr (version 1.0.4)

vaeac_normalize_data: Normalize mixed data for vaeac

Description

Compute the mean and std for each continuous feature, while the categorical features will have mean 0 and std 1.

Usage

vaeac_normalize_data(
  data_torch,
  one_hot_max_sizes,
  norm_mean = NULL,
  norm_std = NULL
)

Value

A list containing the normalized version of x_torch, norm_mean and norm_std.

Arguments

one_hot_max_sizes

A torch tensor of dimension n_features containing the one hot sizes of the n_features features. That is, if the ith feature is a categorical feature with 5 levels, then one_hot_max_sizes[i] = 5. While the size for continuous features can either be 0 or 1.

norm_mean

Torch tensor (optional). A 1D array containing the means of the columns of x_torch.

norm_std

Torch tensor (optional). A 1D array containing the stds of the columns of x_torch.

Author

Lars Henry Berge Olsen