We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 89cc8fe commit 1b62eeeCopy full SHA for 1b62eee
1 file changed
torchTextClassifiers/torchTextClassifiers.py
@@ -223,12 +223,6 @@ def train(
223
X_train, y_train = self._check_XY(X_train, y_train)
224
X_val, y_val = self._check_XY(X_val, y_val)
225
226
- assert (
227
- X_train["categorical_variables"].ndim > 1
228
- and X_train["categorical_variables"].shape[1] == X_val["categorical_variables"].shape[1]
229
- or X_val["categorical_variables"].ndim == 1
230
- )
231
-
232
if (
233
X_train["categorical_variables"] is not None
234
and X_val["categorical_variables"] is not None
0 commit comments