evo_autodecoder: Create a Evolutionary Autodecoder Model

In rdinnager/fibre: Fast Evolutionary Trait Modelling on Phylogenies using Branch Regression Models

Create a Evolutionary Autodecoder Model

Description

Create a Evolutionary Autodecoder Model

Usage

evo_autodecoder(
  latent_dim,
  n_edges,
  decoder,
  reconstruction_loss,
  device,
  decoder_args = list(),
  loss_args = list()
)

Arguments

latent_dim	Number of latent dimensions
decoder	A torch::nn_model() specifying a 'decoder' network architecture. The decoder network should accept a 2 dimensional torch::torch_tensor() with first d

Value

A torch::nn_module()

rdinnager/fibre documentation built on Dec. 14, 2024, 10:33 a.m.