Commit 021a38fb authored by lucas_miranda's avatar lucas_miranda
Browse files

Added tests to account for rule-based label propagation

parent cd692bcb
Pipeline #98196 passed with stages
in 32 minutes and 50 seconds
......@@ -1076,9 +1076,9 @@ class table_dict(dict):
X_train = np.concatenate(list(raw_data))
if self._propagate_labels:
X_train, y_train = X_train[:, :-1], X_train[:, -1]
X_train, y_train = X_train[:, :-1], X_train[:, -1][:, np.newaxis]
try:
X_test, y_test = X_test[:, :-1], X_test[:, -1]
X_test, y_test = X_test[:, :-1], X_test[:, -1][:, np.newaxis]
except IndexError:
pass
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment