def format_data_to_jags()

in pplbench/ppls/jags/robust_regression.py [0:0]

• 6 lines of code
• 4 McCabe index (conditional complexity)

    def format_data_to_jags(self, data: xr.Dataset) -> Dict:
        # transpose the dataset to ensure that it is the way we expect
        data = data.transpose("item", "feature")
        # we already have all the values to be bound except for X and Y in self.attrs
        attrs: dict = self.attrs.copy()
        attrs["X"] = data.X.values
        attrs["Y"] = data.Y.values
        return attrs