rJKOtt package

Submodules

• rJKOtt.DistributionOnGrid module
  • DenseArrayDistribution
    • DenseArrayDistribution.density()
    • DenseArrayDistribution.get_double_moon()
    • DenseArrayDistribution.get_marginal_on_grid()
    • DenseArrayDistribution.get_nonconvex()
    • DenseArrayDistribution.log_density()
  • DistributionOnGrid
    • DistributionOnGrid.density()
    • DistributionOnGrid.dim
    • DistributionOnGrid.get_marginal_on_grid()
    • DistributionOnGrid.log_density()
    • DistributionOnGrid.plot_1d_marginal()
    • DistributionOnGrid.plot_2d_marginal()
    • DistributionOnGrid.plot_matrix_marginals()
  • GaussianMixture
    • GaussianMixture.covariances
    • GaussianMixture.get_marginal_on_grid()
    • GaussianMixture.log_density()
    • GaussianMixture.means
    • GaussianMixture.sample()
    • GaussianMixture.weights
  • Grid
    • Grid.clip_sample()
    • Grid.get_1d_grid()
    • Grid.get_2d_grid()
  • TensorTrainDistribution
    • TensorTrainDistribution.adapt_by_gaussian()
    • TensorTrainDistribution.adapt_by_interpolation()
    • TensorTrainDistribution.adapt_grid()
    • TensorTrainDistribution.density()
    • TensorTrainDistribution.gaussian()
    • TensorTrainDistribution.get_credible_interval()
    • TensorTrainDistribution.get_marginal_on_grid()
    • TensorTrainDistribution.interpolate()
    • TensorTrainDistribution.log_density()
    • TensorTrainDistribution.rank1_fx()
    • TensorTrainDistribution.score()
  • tt_on_grid_grad_of_log()
  • tt_on_grid_interpolate()
  • tt_on_grid_value_and_gradient()
• rJKOtt.TensorTrainSolver module
  • TensorTrainSolver
    • TensorTrainSolver.grid
    • TensorTrainSolver.n_calls
    • TensorTrainSolver.n_cache
    • TensorTrainSolver.Ts
    • TensorTrainSolver.betas
    • TensorTrainSolver.get_current_distribution()
    • TensorTrainSolver.get_intermediate_distribution()
    • TensorTrainSolver.sample()
    • TensorTrainSolver.step()
  • TensorTrainSolverParams
    • TensorTrainSolverParams.cross_n_validation
    • TensorTrainSolverParams.cross_nfev_no_posterior
    • TensorTrainSolverParams.cross_nfev_with_posterior
    • TensorTrainSolverParams.cross_rel_diff
    • TensorTrainSolverParams.cross_use_validation
    • TensorTrainSolverParams.cross_validation_rtol
    • TensorTrainSolverParams.fp_max_iter
    • TensorTrainSolverParams.fp_method
    • TensorTrainSolverParams.fp_relaxation
    • TensorTrainSolverParams.fp_stopping_rtol
    • TensorTrainSolverParams.max_rank
    • TensorTrainSolverParams.max_rank_density
    • TensorTrainSolverParams.max_rank_eta
    • TensorTrainSolverParams.max_rank_hat_eta
    • TensorTrainSolverParams.sampling_n_euler_maruyama_steps
    • TensorTrainSolverParams.sampling_ode_atol
    • TensorTrainSolverParams.sampling_ode_rtol
    • TensorTrainSolverParams.sampling_sde_fraction
    • TensorTrainSolverParams.trunc_tol
    • TensorTrainSolverParams.trunc_tol_density
    • TensorTrainSolverParams.trunc_tol_eta
    • TensorTrainSolverParams.trunc_tol_hat_eta
    • TensorTrainSolverParams.zero_threshold

Module contents