HazyResearch/fonduer

"""Fonduer feature model."""
from sqlalchemy import Column, Float
from sqlalchemy.dialects import postgresql

from fonduer.meta import Meta
from fonduer.utils.models.annotation import AnnotationKeyMixin, AnnotationMixin


class FeatureKey(AnnotationKeyMixin, Meta.Base):
    """A feature's key that identifies the definition of the Feature."""

    pass


class Feature(AnnotationMixin, Meta.Base):
    """An element of a representation of a Candidate in a feature space.

    A Feature's annotation key identifies the definition of the Feature, e.g.,
    a function that implements it or the library name and feature name in an
    automatic featurization library.
    """

    #: A list of floating point values for each Key.
    values = Column(postgresql.ARRAY(Float), nullable=False)