blob: 2db1a833d1a59ad35fed28f683d726c8278ac52d [file]
import numpy as np
from sklearn import datasets
from sklearn import utils
"""
Module to load digit data.
"""
def digit_data() -> utils.Bunch:
return datasets.load_digits()
def target(digit_data: utils.Bunch) -> np.ndarray:
return digit_data.target
def target_names(digit_data: utils.Bunch) -> np.ndarray:
return digit_data.target_names
def feature_matrix(digit_data: utils.Bunch) -> np.ndarray:
# return digit_data.images.reshape((len(digit_data), -1))
return digit_data.data