run_myOT {ROKET}R Documentation

run_myOT

Description

Runs balanced or unbalanced optimal transport on two input vectors

Usage

run_myOT(
  XX,
  YY,
  COST,
  EPS,
  LAMBDA1,
  LAMBDA2 = NULL,
  balance = FALSE,
  conv = 1e-05,
  max_iter = 3000,
  verbose = TRUE,
  show_iter = 50
)

Arguments

XX

A numeric vector of positive masses

YY

A numeric vector of positive masses

COST

A numeric matrix of non-negative values representing the costs to transport masses between features of XX and YY. The rows of COST and features of XX need to be aligned. The columns of COST and features of YY need to be aligned.

EPS

A positive numeric value representing the tuning parameter for entropic regularization.

LAMBDA1

A non-negative numeric value representing the tuning parameter penalizing the distance between XX and the row sums of the optimal transport matrix.

LAMBDA2

A non-negative numeric value representing the tuning parameter penalizing the distance between YY and the column sums of the optimal transport matrix.

balance

Boolean set to TRUE to run balanced optimal transport regardless of LAMDA1 and LAMBDA2. Otherwise run unbalanced optimal transport.

conv

A positive numeric value to determine algorithmic convergence. The default value is 1e-5.

max_iter

A positive integer denoting the maximum iterations to run the algorithm.

verbose

Boolean value to display verbose function output.

show_iter

A positive integer to display iteration details at multiples of show_iter but only if verbose = TRUE.

Value

A R list containing the optimal transport matrix and associated distance metric.


[Package ROKET version 1.0.0 Index]