MLE of Normal parameters

Dataset download


[1]:
import numpy as np
import pandas as pd

Recall our first model for spindle length.

\begin{align} l_i \sim \text{Norm}(\phi, \sigma) \;\;\forall i. \end{align}

For this model, we have already worked out analytically that the maximum likelihood estimates for the parameters \(\phi\) and \(\sigma\) are given by their respective plug-in estimates.

\begin{align} &\phi^* = \hat{\phi}\\[1em] &\sigma^* = \hat{\sigma}. \end{align}

We can directly compute these and use parametric bootstrap to get their confidence intervals. We start by loading in the data and computing the MLEs for \(\phi\) and \(\sigma\).

[2]:
df = pd.read_csv("../data/good_invitro_droplet_data.csv", comment="#")

spindle_length = df["Spindle Length (um)"].values

phi_mle = np.mean(spindle_length)
sigma_mle = np.std(spindle_length)

print("φ MLE:", phi_mle, "µm")
print("σ MLE:", sigma_mle, "µm")
φ MLE: 32.86402985074626 µm
σ MLE: 4.784665043782949 µm

Now we use parametric bootstrap to compute the confidence intervals. First, we’ll write functions to perform the bootstrapping.

[3]:
def draw_parametric_bs_reps(data, phi, sigma, size=1):
    """Parametric bootstrap replicates of parameters of
    Normal distribution."""
    bs_reps_phi = np.empty(size)
    bs_reps_sigma = np.empty(size)

    for i in range(size):
        bs_sample = np.random.normal(phi, sigma, size=len(data))
        bs_reps_phi[i] = np.mean(bs_sample)
        bs_reps_sigma[i] = np.std(bs_sample)

    return bs_reps_phi, bs_reps_sigma

Now we can get the bootstrap replicates.

[4]:
bs_reps_phi, bs_reps_sigma = draw_parametric_bs_reps(
    spindle_length, phi_mle, sigma_mle, size=100000
)

With the bootstrap replicates in hand, we can compute the 95% confidence intervals using percentiles.

[5]:
print('φ conf int:', np.percentile(bs_reps_phi, [2.5, 97.5]))
print('σ conf int:', np.percentile(bs_reps_sigma, [2.5, 97.5]))
φ conf int: [32.50167668 33.22578602]
σ conf int: [4.52462138 5.03646672]

Thus, we have estimated the parameters under this model.

Computing environment

[6]:
%load_ext watermark
%watermark -v -p numpy,pandas,jupyterlab
Python implementation: CPython
Python version       : 3.11.4
IPython version      : 8.12.2

numpy     : 1.24.3
pandas    : 2.0.3
jupyterlab: 4.0.5