Fix LaTeX notation and update print statement

Author: partev

docs/notebooks/linear_gaussian_ssm/lgssm_learning.ipynb

@@ -12,8 +12,8 @@
     "\n",
     "Here, we work with simulated noisy data from an LG-SSM with known parameters, and then we see how well we can recover the true parameters and states given the observations. The model is,\n",
     "\\begin{align*}\n",
-    "z_{t+1} \\mid z_t, \\theta &\\sim \\mathrm{N}(F z_t, Q) \\\\\n",
-    "y_t \\mid z_t, \\theta &\\sim \\mathrm{N}(H z_t, R)\n",
+    "z_{t+1} \\mid z_t, \\theta &\\sim \\mathcal{N}(F z_t, Q) \\\\\n",
+    "y_t \\mid z_t, \\theta &\\sim \\mathcal{N}(H z_t, R)\n",
     "\\end{align*}\n",
     "where $z_{1:T}$ are the latent states, $y_{1:T}$ are the emissions, and $\\theta = (F, Q, H, R)$ are the model parameters. In particular, $F$ is the dynamics matrix and $H$ is the emission matrix. For our simulation, we use 2-dimensional latent states, $z_t \\in \\mathbb{R}^2$, and 10-dimensional emissions, $y_t \\in \\mathbb{R}^{10}$. \n",
     "\n",
@@ -542,7 +542,7 @@
     "sgd_params, sgd_param_props = model.initialize(\n",
     "    init_key, dynamics_weights=true_A, dynamics_covariance=true_Sigma)\n",
     "\n",
-    "print(f\"freeing the dynamics matrix to:\\n {sgd_params.dynamics.weights}\")\n",
+    "print(f\"freezing the dynamics matrix to:\\n {sgd_params.dynamics.weights}\")\n",
     "sgd_param_props.dynamics.weights.trainable = False\n",
     "sgd_param_props.dynamics.cov.trainable = False\n",
     "\n",