diff --git a/cml/ports/source_adapters.py b/cml/ports/source_adapters.py index 1943db14647ed5ffa676256684c6afe9de8b23d7..19222420bdce811c9d08d4d39c6339c671e1c3ad 100644 --- a/cml/ports/source_adapters.py +++ b/cml/ports/source_adapters.py @@ -1,3 +1,4 @@ +from typing import List from abc import abstractmethod, ABC import pandas as pd @@ -59,6 +60,18 @@ class PandasBlock: def __len__(self): return self.__data_block.shape[0] + @property + def labeled(self): + return not self.__data_block.Z.nunique() == 1 + + @property + def columns(self): + return list(self.__data_block.columns) + + @property + def n_features(self): + return self.__data_block.shape[1] - self._LAST_THREE_COLUMNS + def as_numpy_array(self): return self.__data_block[ self.__data_block.columns[