flopsearch 0.2.1__pp310-pypy310_pp73-musllinux_1_2_aarch64.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,25 @@
+import numpy as np
+from typing import Optional
+
+def flop(
+    data: np.ndarray,
+    lambda_bic: float,
+    *,
+    restarts: Optional[int] = None,
+    timeout: Optional[float] = None,
+) -> 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.
+
+    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 (the entry in row j and column i will also be a 2, that is each undirected edge induces two 2's in the matrix).
+    """
+    ...

flopsearch-0.2.1.dist-info/METADATA

@@ -0,0 +1,57 @@
+Metadata-Version: 2.4
+Name: flopsearch
+Version: 0.2.1
+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: Marcel Wienöbst, Sebastian Weichwald, Leonard Henckel
+License-Expression: MPL-2.0
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+Project-URL: Repository, https://github.com/CausalDisco/flopsearch
+Project-URL: Issues, https://github.com/CausalDisco/flopsearch/issues
+
+# 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=50)
+```
+
+## Input and Output
+As input, FLOP takes the data matrix, the BIC penalty parameter (we recommend ```2.0``` as a default choice) and either a ```timeout``` (in seconds) or the number of ILS restarts to control how long the search runs.
+
+The output of FLOP is a CPDAG encoded with an adjacency matrix whose entry in row i and column j is 1 in case of a directed edge from the i-th to the j-th variable and 2 in case of an undirected edge between those variables (in case of an undirected edge, the entry in row j and column i is also 2, that is each undirected edge induces two 2's in the matrix).
+

flopsearch-0.2.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+flopsearch-0.2.1.dist-info/METADATA,sha256=WtG22FLvEokegOCLx0VTFabe3PmXZ7kSaFMxrAaaGUM,2275
+flopsearch-0.2.1.dist-info/WHEEL,sha256=3gtbvScOLf-RZqqhP5oXmH-OzyACot3_IXILT51AfvM,116
+flopsearch.libs/libgcc_s-39080030.so.1,sha256=fIO6GHOh8Ft9CR0Geu7wSUb9Xnl122iTtrxQQ9TAkTQ,789673
+flopsearch/__init__.py,sha256=UqRJYN7saLZWFc0psFyiG39C208VD3OUvagoo2sxfq0,123
+flopsearch/__init__.pyi,sha256=seVc-AOZrar0YHazomEwlN6rVD_vDzOiDDnJ3rlBgjA,1119
+flopsearch/flopsearch.pypy310-pp73-aarch64-linux-gnu.so,sha256=_9eeLRtQqACMuBDhN2QpIQxZDv5E9RDPGWqnrSNHGaI,1051545
+flopsearch/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+flopsearch-0.2.1.dist-info/RECORD,,

flopsearch-0.2.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: maturin (1.10.2)
+Root-Is-Purelib: false
+Tag: pp310-pypy310_pp73-musllinux_1_2_aarch64