From 0ee66d4d082cea8647d86264f3bc33ad277d420c Mon Sep 17 00:00:00 2001
From: dmt <>
Date: Sun, 20 Oct 2019 23:18:00 +0200
Subject: [PATCH] Provide property for checking if the data block has labels or
not.
---
cml/ports/source_adapters.py | 13 +++++++++++++
1 file changed, 13 insertions(+)
diff --git a/cml/ports/source_adapters.py b/cml/ports/source_adapters.py
index 1943db1..1922242 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[
--
GitLab