sktime_knn_time_series_classifier_with_dtw_distance.py

1from sktime.datasets import load_unit_test
2from sktime.classification.distance_based import KNeighborsTimeSeriesClassifier
3
4# Load the Unit Test dataset
5X_train, y_train = load_unit_test(split="train", return_X_y=True)
6X_test, y_test = load_unit_test(split="test", return_X_y=True)
7
8# Initialize and fit the classifier
9clf = KNeighborsTimeSeriesClassifier(distance="dtw")
10clf.fit(X_train, y_train)
11
12# Predict on the test set and calculate accuracy
13y_pred = clf.predict(X_test)
14accuracy = clf.score(X_test, y_test)
15
16print(f"Predicted labels: {y_pred[:5]}")
17print(f"Accuracy: {accuracy}")