Upload FIMMJP

mjp.py

@@ -159,10 +159,13 @@ class FIMMJP(AModel):
 
         out = {
             "intensity_matrices": create_matrix_from_off_diagonal(
-                pred_offdiag_im_mean, self.n_states, mode="sum_row", n_states=self.n_states if n_states is None else n_states
+                pred_offdiag_im_mean, self.n_states, mode="negative_sum_row", n_states=self.n_states if n_states is None else n_states
             ),
             "intensity_matrices_variance": create_matrix_from_off_diagonal(
-                torch.exp(pred_offdiag_im_logvar), self.n_states, mode="sum_row", n_states=self.n_states if n_states is None else n_states
+                torch.exp(pred_offdiag_im_logvar),
+                self.n_states,
+                mode="negative_sum_row",
+                n_states=self.n_states if n_states is None else n_states,
             ),
             "initial_condition": init_cond,
         }