executing-plans
executing-plans
Use when executing an implementation plan in the current session as the implementer yourself — your human partner chose inline execution, or no subagent tool is available
ds-maximum-likelihoodImplements maximum likelihood estimation, likelihood functions, and optimization methods for parameter estimation in probabilistic models
Independent GitHub snapshot · Automated classification · Original documentation governs
name: ds-maximum-likelihood compatibility: opencode completeness: 95 content-types:
Comprehensive guide to maximum likelihood estimation in machine learning and data science workflows.
Maximum Likelihood Estimation is a critical component of the machine learning workflow. This skill covers:
import numpy as np
from scipy.optimize import minimize
from scipy.stats import norm
def negative_log_likelihood(params: np.ndarray, data: np.ndarray) -> float:
"""Calculate negative log-likelihood for Gaussian distribution."""
mu, sigma = params
if sigma <= 0:
return np.inf
return -np.sum(norm.logpdf(data, loc=mu, scale=sigma))
def basic_mle(data: np.ndarray) -> dict:
"""
Perform basic Maximum Likelihood Estimation for Gaussian parameters.
Args:
data: 1D array of observations
Returns:
Dictionary containing estimated parameters and optimization status
"""
if data is None or len(data) == 0:
raise ValueError("Input data cannot be empty")
initial_guess = np.array([np.mean(data), np.std(data)])
result = minimize(
negative_log_likelihood,
initial_guess,
args=(data,),
method='Nelder-Mead'
)
if not result.success:
raise RuntimeError(f"MLE optimization failed: {result.message}")
return {
'mu': result.x[0]
'sigma': result.x[1]
'log_likelihood': -result.fun
'converged': result.success
}
# Generate sample data and run estimation
np.random.seed(42)
sample_data = np.random.normal(loc=5.0, scale=2.0, size=1000)
mle_results = basic_mle(sample_data)
print(f"Estimated mu: {mle_results['mu']:.4f}, sigma: {mle_results['sigma']:.4f}")
import logging
import pandas as pd
import numpy as np
from scipy.optimize import minimize
from scipy.stats import norm
from typing import Any, Dict, List
logger = logging.getLogger(__name__)
class MaximumLikelihoodEstimator:
"""Production implementation of Maximum Likelihood Estimation for Gaussian distributions."""
def __init__(self, method: str = 'Nelder-Mead', tol: float = 1e-6):
self.method = method
self.tol = tol
self.results_: Dict[str, Any] = {}
def _negative_log_likelihood(self, params: List[float], data: np.ndarray) -> float:
mu, sigma = params
if sigma <= 0:
return np.inf
return -np.sum(np.log(norm.pdf(data, loc=mu, scale=sigma)))
def fit(self, data: pd.DataFrame, column: str) -> 'MaximumLikelihoodEstimator':
if column not in data.columns:
raise ValueError(f"Column '{column}' not found in DataFrame")
values = data[column].dropna().values
if len(values) == 0:
raise ValueError("No valid data points found in specified column")
initial_guess = [np.mean(values), np.std(values)]
result = minimize(
self._negative_log_likelihood,
initial_guess,
args=(values,),
method=self.method
tol=self.tol
)
if not result.success:
logger.warning(f"MLE optimization did not converge: {result.message}")
self.results_ = {
'parameters': {'mu': result.x[0], 'sigma': result.x[1]}
'log_likelihood': -result.fun
'converged': result.success
'iterations': result.nit
}
return self
def execute(self, data: pd.DataFrame) -> Dict[str, Any]:
"""Return the fitted results. Requires fit() to be called first."""
if not self.results_:
raise RuntimeError("Model has not been fitted. Call fit() first.")
return self.results_
# BAD: Repeated likelihood calculations, no error handling, magic numbers
def bad_mle(data):
mu = sum(data) / len(data)
sigma = sum((x - mu)**2 for x in data) / len(data)
return mu, sigma # Ignores optimization, uses MLE closed-form incorrectly for small samples
# GOOD: Modular, validated, uses scipy optimizer, follows DRY principle
def good_mle(data: np.ndarray) -> Dict[str, float]:
if len(data) < 2:
raise ValueError("Need at least 2 points for reliable estimation")
def nll(p):
return -np.sum(norm.logpdf(data, loc=p[0], scale=p[1]))
res = minimize(nll, [np.mean(data), np.std(data)], method='L-BFGS-B')
return {'mu': res.x[0], 'sigma': res.x[1], 'll': -res.fun}
| Pitfall | Problem | Solution | |
Authoritative documentation links for this skill's domain. The model follows markdown links at load time to resolve external references and inline content.