Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 2 additions & 3 deletions muon/_core/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -370,7 +370,7 @@ def mofa(
use reduced precision (float32)
gpu_mode : optional
if to use GPU mode
gpu_mode : optional
gpu_device : optional
which GPU device to use
svi_mode : optional
if to use Stochastic Variational Inference (SVI)
Expand Down Expand Up @@ -404,8 +404,7 @@ def mofa(
path to HDF5 file to store the model
expectations : optional
which nodes should be used to save expectations for (will save only W and Z by default);
possible expectations names
nclude Y, W, Z, Tau, AlphaZ, AlphaW, ThetaW, ThetaZ
possible expectation names include Y, W, Z, Tau, AlphaZ, AlphaW, ThetaW, ThetaZ
save_interrupted : optional
if to save partially trained model when the training is interrupted
verbose : optional
Expand Down