febolt 0.1.59__tar.gz → 0.1.60__tar.gz

{febolt-0.1.59 → febolt-0.1.60}/Cargo.lock

@@ -362,7 +362,7 @@ checksum = "37909eebbb50d72f9059c3b6d82c0463f2ff062c9e95845c43a6c9c0355411be"
 
 [[package]]
 name = "febolt"
-version = "0.1.59"
+version = "0.1.60"
 dependencies = [
  "blas",
  "intel-mkl-src",

{febolt-0.1.59 → febolt-0.1.60}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "febolt"
-version = "0.1.59"
+version = "0.1.60"
 edition = "2021"
 description = "Statistics library for Python powered by Rust"
 license = "MIT"

{febolt-0.1.59 → febolt-0.1.60}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: febolt
-Version: 0.1.59
+Version: 0.1.60
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy

{febolt-0.1.59 → febolt-0.1.60}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "febolt"
-version = "0.1.59"
+version = "0.1.60"
 requires-python = ">=3.8"
 description = "A Rust-based Statistics and ML package, callable from Python."
 keywords = ["rust", "python", "Machine Learning", "Statistics", "pyo3"]

{febolt-0.1.59 → febolt-0.1.60}/src/lib.rs

@@ -295,6 +295,7 @@ fn ame<'py>(
     // 10) Build final results
     let (mut dy_dx, mut se_err, mut z_vals, mut p_vals, mut sig) =
         (Vec::new(), Vec::new(), Vec::new(), Vec::new(), Vec::new());
+    let (mut conf_low, mut conf_high) = (Vec::new(), Vec::new());
     let mut names_out = Vec::new();
 
     for j in 0..k {
@@ -311,21 +312,28 @@ fn ame<'py>(
             z_vals.push(z);
             p_vals.push(p);
             sig.push(add_significance_stars(p));
+            // Compute 95% confidence interval using 1.96 * SE
+            conf_low.push(dy - 1.96 * s);
+            conf_high.push(dy + 1.96 * s);
         } else {
             z_vals.push(0.0);
             p_vals.push(1.0);
             sig.push("");
+            conf_low.push(dy);
+            conf_high.push(dy);
         }
         names_out.push(exog_names[j].clone());
     }
 
-    // 11) Create DataFrame
+    // 11) Create DataFrame with extra confidence interval columns.
     let pd = py.import("pandas")?;
     let data = PyDict::new(py);
     data.set_item("dy/dx", &dy_dx)?;
     data.set_item("Std. Err", &se_err)?;
     data.set_item("z", &z_vals)?;
     data.set_item("Pr(>|z|)", &p_vals)?;
+    data.set_item("Conf. Int. Low", &conf_low)?;
+    data.set_item("Conf. Int. Hi", &conf_high)?;
     data.set_item("Significance", &sig)?;
 
     let kwargs = PyDict::new(py);
@@ -341,3 +349,4 @@ fn febolt(_py: Python, m: &PyModule) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(ame, m)?)?;
     Ok(())
 }
+