Skip to content

Commit 652db10

Browse files
committed
Update LightGBM with validation data
1 parent 6a315d2 commit 652db10

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

autoemulate/emulators/lightgbm.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,12 @@ def is_multioutput() -> bool:
157157
"""LightGBM does not support multi-output."""
158158
return False
159159

160-
def _fit(self, x: TensorLike, y: TensorLike): # type: ignore since this is valid subclass of types
160+
def _fit(
161+
self,
162+
x: TensorLike,
163+
y: TensorLike,
164+
validation_data: tuple[TensorLike, TensorLike] | None = None, # noqa: ARG002
165+
) -> None:
161166
x_np, y_np = self._convert_to_numpy(x, y)
162167
self.n_features_in_ = x_np.shape[1]
163168
self.model_.fit(x_np, y_np)

0 commit comments

Comments
 (0)