flopsearch 0.1.3__cp313-cp313-musllinux_1_2_i686.whl

flopsearch/__init__.py

@@ -0,0 +1,5 @@
+from .flopsearch import *
+
+__doc__ = flopsearch.__doc__
+if hasattr(flopsearch, "__all__"):
+    __all__ = flopsearch.__all__

flopsearch/__init__.pyi

@@ -0,0 +1,26 @@
+import numpy as np
+
+def flop(
+    data: np.ndarray,
+    lambda_bic: float,
+    *,
+    restarts: int,
+    timeout: float,
+    output_dag: bool = False,
+) -> np.ndarray:
+    """
+    Run the FLOP causal discovery algorithm.
+
+    Parameters
+    ----------
+    data: A data matrix with rows corresponding to observations and columns to variables/nodes.
+    lambda_bic: The penalty parameter of the BIC, a typical value for structure learning is 2.0.
+    restarts: Optional parameter specifying the number of ILS restarts. Either restarts or timeout (below) need to be specified.
+    timeout: Optional parameter specifying a timeout after which the search returns. At least one local search is run up to a local optimum. Either restarts or timeout need to be specified.
+    output_dag: Optional parameter to output a DAG instead of a CPDAG. Default value is False.
+
+    Returns
+    -------
+    A matrix encoding a CPDAG or DAG. The entry in row i and column j is 1 in case of a directed edge from i to j and 2 in case of an undirected edge between those nodes (an additional 2 in row j and column i is omitted).
+    """
+    ...

flopsearch-0.1.3.dist-info/METADATA

@@ -0,0 +1,50 @@
+Metadata-Version: 2.4
+Name: flopsearch
+Version: 0.1.3
+Classifier: Programming Language :: Rust
+Classifier: Programming Language :: Python :: Implementation :: CPython
+Classifier: Programming Language :: Python :: Implementation :: PyPy
+Summary: Python package providing an implementation of the FLOP causal discovery algorithm for linear additive noise models
+Keywords: causal discovery,DAGs,Bayesian networks,structure learning
+Author: Sebastian Weichwald, Leonard Henckel
+Author-email: Marcel Wienöbst <marcel.wienoebst@gmx.de>
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+
+# flopsearch
+
+Python package providing an implementation of the FLOP causal discovery algorithm for linear additive noise models.
+
+## Installation
+flopsearch can be installed via pip:
+
+```bash
+pip install flopsearch
+```
+
+## Citing FLOP
+If you use FLOP in your scientific work, please cite this paper:
+```bibtex
+@article{cifly2025,
+  author  = {Marcel Wien{"{o}}bst and Leonard Henckel and Sebastian Weichwald},
+  title   = {{Embracing Discrete Search: A Reasonable Approach to Causal Structure Learning}},
+  journal = {{arXiv preprint arXiv:2510.04970}},
+  year    = {2025}
+}
+```
+
+## Example
+A simple example run of the FLOP algorithm provided by flopsearch.
+
+``` py
+import flopsearch
+import numpy as np
+from scipy import linalg
+
+p = 10
+W = np.diag(np.ones(p - 1), 1)
+X = np.random.randn(10000, p).dot(linalg.inv(np.eye(p) - W))
+X_std = (X - np.mean(X, axis=0)) / np.std(X, axis=0)
+flopsearch.flop(X_std, 2.0, restarts=20)
+```
+

flopsearch-0.1.3.dist-info/RECORD

@@ -0,0 +1,8 @@
+flopsearch-0.1.3.dist-info/METADATA,sha256=JVX_fvtqJZGFRQayG9XLtnhLFBg3QgLaBsfRqIi7NsY,1551
+flopsearch-0.1.3.dist-info/WHEEL,sha256=hnEjxG8MMZmmlhJat41pJlHVa7PMz9kEhUjTj-jdJr4,105
+flopsearch.libs/libgcc_s-27e5a392.so.1,sha256=x5sO63liVwXxrjGGP371wB0RyQe1KEnIynYm82T0G0M,449745
+flopsearch/__init__.py,sha256=UqRJYN7saLZWFc0psFyiG39C208VD3OUvagoo2sxfq0,123
+flopsearch/__init__.pyi,sha256=yQdOZROGWM0k0sdNj-KiielkkQcxiAcEgxaunbqJw9E,1122
+flopsearch/flopsearch.cpython-313-i386-linux-musl.so,sha256=01lzXI8X-I4ZuugkwA8F31Va0EIoTSMX2gn1tGUNdtw,805597
+flopsearch/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+flopsearch-0.1.3.dist-info/RECORD,,

flopsearch-0.1.3.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: maturin (1.9.6)
+Root-Is-Purelib: false
+Tag: cp313-cp313-musllinux_1_2_i686