an class="n">muse_fit_predict</span><span class="p">(</span><span class="n">data_x</span><span class="p">,</span>
                                                  <span class="n">data_y</span><span class="p">,</span>
                                                  <span class="n">label_x</span><span class="p">,</span>
                                                  <span class="n">label_y</span><span class="p">,</span>
                                                  <span class="n">latent_dim</span><