blob: bbd640723f4601b25f5056216f569272153a27a7 [file] [log] [blame]
import numpy as np
from sklearn import datasets, 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