qxmt.datasets.schema module

qxmt.datasets.schema module#

class qxmt.datasets.schema.Dataset(*, X_train, y_train, X_val, y_val, X_test, y_test, config)

Bases: BaseModel

Parameters:

  • X_train (Annotated[ndarray, PlainValidator(func=~qxmt.datasets.schema.validate, json_schema_input_type=~typing.Any), PlainSerializer(func=~qxmt.datasets.schema.serialize, return_type=PydanticUndefined, when_used=always)])

  • y_train (Annotated[ndarray, PlainValidator(func=~qxmt.datasets.schema.validate, json_schema_input_type=~typing.Any), PlainSerializer(func=~qxmt.datasets.schema.serialize, return_type=PydanticUndefined, when_used=always)])

  • X_val (Annotated[ndarray, PlainValidator(func=~qxmt.datasets.schema.validate, json_schema_input_type=~typing.Any), PlainSerializer(func=~qxmt.datasets.schema.serialize, return_type=PydanticUndefined, when_used=always)] | None)

  • y_val (Annotated[ndarray, PlainValidator(func=~qxmt.datasets.schema.validate, json_schema_input_type=~typing.Any), PlainSerializer(func=~qxmt.datasets.schema.serialize, return_type=PydanticUndefined, when_used=always)] | None)

  • X_test (Annotated[ndarray, PlainValidator(func=~qxmt.datasets.schema.validate, json_schema_input_type=~typing.Any), PlainSerializer(func=~qxmt.datasets.schema.serialize, return_type=PydanticUndefined, when_used=always)])

  • y_test (Annotated[ndarray, PlainValidator(func=~qxmt.datasets.schema.validate, json_schema_input_type=~typing.Any), PlainSerializer(func=~qxmt.datasets.schema.serialize, return_type=PydanticUndefined, when_used=always)])

  • config (DatasetConfig)

X_test: Annotated[ndarray, PlainValidator(func=validate, json_schema_input_type=Any), PlainSerializer(func=serialize, return_type=PydanticUndefined, when_used=always)]
X_train: Annotated[ndarray, PlainValidator(func=validate, json_schema_input_type=Any), PlainSerializer(func=serialize, return_type=PydanticUndefined, when_used=always)]
X_val: Annotated[ndarray, PlainValidator(func=validate, json_schema_input_type=Any), PlainSerializer(func=serialize, return_type=PydanticUndefined, when_used=always)] | None
config: DatasetConfig
model_computed_fields: ClassVar[Dict[str, ComputedFieldInfo]] = {}

A dictionary of computed field names and their corresponding ComputedFieldInfo objects.

model_config: ClassVar[ConfigDict] = {}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].

model_fields: ClassVar[Dict[str, FieldInfo]] = {'X_test': FieldInfo(annotation=ndarray, required=True, metadata=[PlainValidator(func=<function validate>, json_schema_input_type=typing.Any), PlainSerializer(func=<function serialize>, return_type=PydanticUndefined, when_used='always')]), 'X_train': FieldInfo(annotation=ndarray, required=True, metadata=[PlainValidator(func=<function validate>, json_schema_input_type=typing.Any), PlainSerializer(func=<function serialize>, return_type=PydanticUndefined, when_used='always')]), 'X_val': FieldInfo(annotation=Union[Annotated[ndarray, PlainValidator, PlainSerializer], NoneType], required=True), 'config': FieldInfo(annotation=DatasetConfig, required=True), 'y_test': FieldInfo(annotation=ndarray, required=True, metadata=[PlainValidator(func=<function validate>, json_schema_input_type=typing.Any), PlainSerializer(func=<function serialize>, return_type=PydanticUndefined, when_used='always')]), 'y_train': FieldInfo(annotation=ndarray, required=True, metadata=[PlainValidator(func=<function validate>, json_schema_input_type=typing.Any), PlainSerializer(func=<function serialize>, return_type=PydanticUndefined, when_used='always')]), 'y_val': FieldInfo(annotation=Union[Annotated[ndarray, PlainValidator, PlainSerializer], NoneType], required=True)}

Metadata about the fields defined on the model, mapping of field names to [FieldInfo][pydantic.fields.FieldInfo] objects.

This replaces Model.__fields__ from Pydantic V1.

y_test: Annotated[ndarray, PlainValidator(func=validate, json_schema_input_type=Any), PlainSerializer(func=serialize, return_type=PydanticUndefined, when_used=always)]
y_train: Annotated[ndarray, PlainValidator(func=validate, json_schema_input_type=Any), PlainSerializer(func=serialize, return_type=PydanticUndefined, when_used=always)]
y_val: Annotated[ndarray, PlainValidator(func=validate, json_schema_input_type=Any), PlainSerializer(func=serialize, return_type=PydanticUndefined, when_used=always)] | None
qxmt.datasets.schema.serialize(v)
Parameters:

v (ndarray)

Return type:

list[list[float]]

qxmt.datasets.schema.validate(v)
Parameters:

v (Any)

Return type:

ndarray