Added interface and module for ccb

estimators/ccb/base.py

@@ -0,0 +1,51 @@
+""" Interface for implementation of conditional contextual bandits estimators """
+
+from abc import ABC, abstractmethod
+from typing import List
+
+class Estimator(ABC):
+	""" Interface for implementation of conditional contextual bandits estimators """
+
+	@abstractmethod
+	def add_example(self, p_log: List, r: List, p_pred: List, count: float) -> None:
+		""" 
+		Args:
+			p_log: List of probability of the logging policy
+			r: List of reward for choosing an action in the given context
+			p_pred: List of predicted probability of making decision
+			count: weight
+		"""
+		...
+
+	@abstractmethod
+	def get(self) -> float:
+		""" Calculates the selected estimator
+
+		Returns:
+			The estimator value
+		"""
+		...
+
+class Interval(ABC):
+	""" Interface for implementation of conditional contextual bandits estimators interval """
+
+	@abstractmethod
+	def add_example(self, p_log: List, r: List, p_pred: List, count: float) -> None:
+		""" 
+		Args:
+			p_log: List of probability of the logging policy
+			r: List of reward for choosing an action in the given context
+			p_pred: List of predicted probability of making decision
+			count: weight
+		"""
+		...
+
+	@abstractmethod
+	def get(self, alpha: float) -> List:
+		""" Calculates the CI
+		Args:
+			alpha: alpha value
+		Returns:
+			Returns the confidence interval as list[float]
+		"""
+		...

setup.py

@@ -18,7 +18,7 @@
         "Operating System :: OS Independent",
         "Topic :: Scientific/Engineering"
     ],
-    packages=["estimators", "estimators.bandits", "estimators.slates", "estimators.utils"],
+    packages=["estimators", "estimators.bandits", "estimators.ccb", "estimators.slates", "estimators.utils"],
     install_requires= ['scipy>=0.9'],
     tests_require=['pytest'],
     python_requires=">=3.6",