KNN Dataset

knn_dataset.py

from sklearn.datasets import load_digits
import matplotlib.pyplot as plt

# Load data set and corresponding labels
digits_data = load_digits()
digits = digits_data.data
labels = digits_data.target
class_names = digits_data.target_names

# Partition data set into training and test data sets
train_data_ratio = 0.75  # test_data_ratio = 1 - train_data_ratio
train_data_idx = int(train_data_ratio * digits.shape[0])

train_data = digits[:train_data_idx]
train_data_labels = labels[:train_data_idx]

test_data = digits[train_data_idx:]
test_data_labels = labels[train_data_idx:]