Code / scripts/ml_training/essay_scoring/features/combiner.py

scripts/ml_training/essay_scoring/features/combiner.py

View this file on Codeberg

  1
  """Feature-matrix result contract for essay scoring."""


  2
  


  3
  from __future__ import annotations


  4
  


  5
  from dataclasses import dataclass


  6
  


  7
  import numpy as np


  8
  


  9
  


  10
  @dataclass(frozen=True)


  11
  class FeatureMatrix:


  12
      """Container for feature matrix and ordered feature names."""


  13
  


  14
      matrix: np.ndarray


  15
      feature_names: list[str]