caikit.interfaces.common.data_model.vectors

Data structures for embedding vector representations

Attributes

log

error

Classes

PyFloatSequence

A DataObject is a data model class that is backed by a @dataclass.

NpFloat32Sequence

A DataObject is a data model class that is backed by a @dataclass.

NpFloat64Sequence

A DataObject is a data model class that is backed by a @dataclass.

Vector1D

Data representation for a 1 dimension vector of float-type data.

ListOfVector1D

Data representation for an embedding matrix holding 2D vectors

Module Contents

caikit.interfaces.common.data_model.vectors.log[source]
caikit.interfaces.common.data_model.vectors.error
class caikit.interfaces.common.data_model.vectors.PyFloatSequence[source]

Bases: caikit.core.DataObjectBase

A DataObject is a data model class that is backed by a @dataclass.

Data model classes that use the @dataobject decorator must derive from this base class.

values: py_to_proto.dataclass_to_proto.Annotated[List[float], FieldNumber(1)] = []
class caikit.interfaces.common.data_model.vectors.NpFloat32Sequence[source]

Bases: caikit.core.DataObjectBase

A DataObject is a data model class that is backed by a @dataclass.

Data model classes that use the @dataobject decorator must derive from this base class.

values: py_to_proto.dataclass_to_proto.Annotated[List[numpy.float32], FieldNumber(1)]
classmethod from_proto(proto)[source]

Build a DataBase from protobufs.

Args:

proto: A protocol buffer to serialize from.

Returns:

protobufs: A DataBase object.

class caikit.interfaces.common.data_model.vectors.NpFloat64Sequence[source]

Bases: caikit.core.DataObjectBase

A DataObject is a data model class that is backed by a @dataclass.

Data model classes that use the @dataobject decorator must derive from this base class.

values: py_to_proto.dataclass_to_proto.Annotated[List[numpy.float64], FieldNumber(1)]
classmethod from_proto(proto)[source]

Build a DataBase from protobufs.

Args:

proto: A protocol buffer to serialize from.

Returns:

protobufs: A DataBase object.

class caikit.interfaces.common.data_model.vectors.Vector1D[source]

Bases: caikit.core.DataObjectBase

Data representation for a 1 dimension vector of float-type data.

data: py_to_proto.dataclass_to_proto.Annotated[PyFloatSequence | NpFloat32Sequence | NpFloat64Sequence, FieldNumber(1)]
__post_init__()[source]
classmethod from_vector(vector)[source]
classmethod from_json(json_str: Dict[str, Any] | str) Vector1D[source]

JSON does not have different float types. Move data into data_pyfloatsequence

to_dict() dict[source]

to_dict is needed to make things serializable

classmethod from_proto(proto)[source]

Wrap the data in an appropriate float sequence, wrapped by this class

fill_proto(proto)[source]

Fill in the data in an appropriate data_<float type sequence>

class caikit.interfaces.common.data_model.vectors.ListOfVector1D[source]

Bases: caikit.core.DataObjectBase

Data representation for an embedding matrix holding 2D vectors

vectors: py_to_proto.dataclass_to_proto.Annotated[List[Vector1D], FieldNumber(1)]
__post_init__()[source]
classmethod from_json(json_str: Dict[str, Any] | str) ListOfVector1D[source]

Fill in the vector data in an appropriate data_<float type sequence>