Fix tiling

Author: pockerman

src/datasets/datasets_loaders.py

@@ -15,7 +15,7 @@
 class MockSubjectsData(object):
 
     # Path to the dataset file
-    FILENAME: Path = Path("../../data/mocksubjects.csv")
+    FILENAME: Path = Path("/home/alex/qi3/drl_anonymity/data/mocksubjects.csv") #("../../data/mocksubjects.csv")
 
     # the assumed column types. We use this map to cast
     # the types of the columns
@@ -57,7 +57,7 @@ def from_options(cls, *, filename: Path,
                                 NORMALIZED_COLUMNS=column_normalization)
         return cls(data=data)
 
-    def __init__(self, data: MockSubjectsData, do_read: bool=True):
+    def __init__(self, data: MockSubjectsData, do_read: bool = True):
         super(MockSubjectsLoader, self).__init__(columns=data.COLUMNS_TYPES)
 
         if do_read:

src/spaces/action_space.py

@@ -10,9 +10,15 @@
 
 
 class ActionSpace(Discrete):
+    """ActionSpace class models a discrete action space of size n
     """
-    ActionSpace class models a discrete action space of size n
-    """
+
+    @classmethod
+    def from_actions(cls, *actions: ActionBase):
+
+        space = cls(n=len(actions))
+        space.add_many(*actions)
+        return space
 
     def __init__(self, n: int) -> None:
 

src/spaces/discrete_state_environment.py

@@ -62,6 +62,15 @@ def from_options(cls, *, data_set: DataSet, action_space: ActionSpace,
                                    punish_factor=punish_factor, max_distortion=max_distortion, gamma=gamma,
                                    n_states=n_states, min_distortion=min_distortion,
                                    average_distortion_constraint=average_distortion_constraint)
+
+        return cls(env_config=config)
+
+    @classmethod
+    def from_dataset(cls, data_set: DataSet, *,  action_space: ActionSpace=None,
+                     reward_manager: RewardManager = None, distortion_calculator: DistortionCalculator = None):
+
+        config = DiscreteEnvConfig(data_set=data_set, action_space=action_space, reward_manager=reward_manager,
+                                   distortion_calculator=distortion_calculator)
         return cls(env_config=config)
 
     def __init__(self, env_config: DiscreteEnvConfig) -> None:

src/spaces/state.py

@@ -4,6 +4,7 @@
 
 """
 
+import numpy as np
 from typing import TypeVar, List, Any
 from src.exceptions.exceptions import Error
 
@@ -65,12 +66,12 @@ def __init__(self):
         self.column_distortions = {}
 
     def __contains__(self, column_name: str) -> bool:
-        """
-        Returns true if column_name is in the column_distortions
+        """Returns true if column_name is in the column_distortions
         keys
 
         Parameters
         ----------
+
         column_name: The column name to query
 
         Returns
@@ -85,11 +86,11 @@ def __iter__(self):
         return StateIterator(list(self.column_distortions.keys()))
 
     def __getitem__(self, name: str) -> float:
-        """
-        Get the distortion corresponding to the name-th column
+        """Get the distortion corresponding to the name-th column
 
         Parameters
         ----------
+
         name: The name of the column
 
         Returns
@@ -99,4 +100,15 @@ def __getitem__(self, name: str) -> float:
         """
         return self.column_distortions[name]
 
+    def to_numpy(self) -> np.array:
+        """Returns the self.column_distortions values as numpy array
+
+        Returns
+        -------
+            np.array
+
+        """
+
+        vals = self.column_distortions.values()
+        return np.array(vals)