tfprobability: layer_one_hot_categorical – R documentation

Pricing

Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!

Get Started for Free

Documentation

tfprobability

layer_one_hot_categorical

A d-variate OneHotCategorical Keras layer from d params.

Description

Typical choices for convert_to_tensor_fn include:

tfp$distributions$Distribution$sample
tfp$distributions$Distribution$mean
tfp$distributions$Distribution$mode
tfp$distributions$OneHotCategorical$logits

Usage

layer_one_hot_categorical(
  object,
  event_size,
  convert_to_tensor_fn = tfp$distributions$Distribution$sample,
  sample_dtype = NULL,
  validate_args = FALSE,
  ...
)

Arguments

`object`	Model or layer object
`event_size`	Scalar `integer` representing the size of single draw from this distribution.
`convert_to_tensor_fn`	A callable that takes a tfd$Distribution instance and returns a tf$Tensor-like object. Default value: `tfd$distributions$Distribution$sample`.
`sample_dtype`	`dtype` of samples produced by this distribution. Default value: `NULL` (i.e., previous layer's `dtype`).
`validate_args`	Logical, default FALSE. When TRUE distribution parameters are checked for validity despite possibly degrading runtime performance. When FALSE invalid inputs may silently render incorrect outputs. Default value: FALSE.
`...`	Additional arguments passed to `args` of `keras::create_layer`.

Value

a Keras layer