Is there a way to enable early stopping callback for .fit() functions?

[jfarshchi]

Hi, Currently, I am using the TimeSeriesKMeans and trying to fit and predict with soft_dtw and the while loop in the fit function goes on and does not stop. I was wondering if there is a way to enable an early stopping callback or a similar alternative for the fitting to stop.

The .fit() function that I am referring to is found below:
https://github.com/tslearn-team/tslearn/blob/9937946/tslearn/clustering/kmeans.py#L764-L831

Here is my model config:

km_soft_dtw = TimeSeriesKMeans(n_clusters=10, n_init=2, metric="softdtw", verbose=True, max_iter_barycenter=10, random_state=seed)

y_pred_soft_dtw = km_soft_dtw .fit(X_train)

Thank you in advance.