From 71e384a741e52f94882b14062a3dc10e5f391533 Mon Sep 17 00:00:00 2001
From: Cathy Yeh <cathy@driver.xyz>
Date: Mon, 20 Nov 2017 11:40:02 -0800
Subject: BernoulliOrCPD inherits from TabularCPD

---
 beliefs/factors/CPD.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 beliefs/factors/CPD.py

(limited to 'beliefs/factors/CPD.py')

diff --git a/beliefs/factors/CPD.py b/beliefs/factors/CPD.py
new file mode 100644
index 0000000..8de47b3
--- /dev/null
+++ b/beliefs/factors/CPD.py
@@ -0,0 +1,36 @@
+import numpy as np
+
+
+class TabularCPD:
+    """
+    Defines the conditional probability table for a discrete variable
+    whose parents are also discrete.
+
+    TODO: have this inherit from DiscreteFactor
+    """
+    def __init__(self, variable, variable_card,
+                 parents=[], parents_card=[], values=[]):
+        """
+        Args:
+          variable: int or string
+          variable_card: int
+          parents: optional, list of int and/or strings
+          parents_card: optional, list of int
+          values: optional, 2d list or array
+        """
+        self.variable = variable
+        self.parents = parents
+        self.variables = [variable] + parents
+        self.cardinality = [variable_card] + parents_card
+
+        if values:
+            self.values = np.array(values)
+
+    def get_values(self):
+        """
+        Returns the tabular cpd form of the values.
+        """
+        if len(self.cardinality) == 1:
+            return self.values.reshape(1, np.prod(self.cardinality))
+        else:
+            return self.values.reshape(self.cardinality[0], np.prod(self.cardinality[1:]))
-- 
cgit v1.2.3