From 77d8b323d4f6e05ca97d9cbef43ac85fd8040d61 Mon Sep 17 00:00:00 2001
From: Cathy Yeh <cathy@driver.xyz>
Date: Mon, 13 Nov 2017 14:42:52 -0800
Subject: copy scripts from lgs branch

---
 beliefs/models/BernoulliOrModel.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)
 create mode 100644 beliefs/models/BernoulliOrModel.py

(limited to 'beliefs/models/BernoulliOrModel.py')

diff --git a/beliefs/models/BernoulliOrModel.py b/beliefs/models/BernoulliOrModel.py
new file mode 100644
index 0000000..da18fb6
--- /dev/null
+++ b/beliefs/models/BernoulliOrModel.py
@@ -0,0 +1,17 @@
+from beliefs.models.BayesianModel import BayesianModel
+from beliefs.types.BernoulliOrNode import BernoulliOrNode
+
+
+class BernoulliOrModel(BayesianModel):
+    """
+    BernoulliOrModel stores node instances of BernoulliOrNodes (Bernoulli
+    variables associated with an OR conditional probability distribution).
+    """
+    def __init__(self, edges, nodes=None):
+        """
+        Input:
+          edges: an edge list, e.g. [(parent1, child1), (parent1, child2)]
+        """
+        if nodes is None:
+            nodes = self.create_nodes(edges, node_class=BernoulliOrNode)
+        super().__init__(edges, nodes=nodes)
-- 
cgit v1.2.3