add_model_resnet18 #31
@@ -0,0 +1,14 @@
|
||||
"""data_models module content."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .angle import AngleData
|
||||
from .contact import ContactData
|
||||
from .distance import DistanceData
|
||||
from .framing import FramingData
|
||||
from .information_value import InformationValueData
|
||||
from .modality_color import ModalityColorData
|
||||
from .modality_depth import ModalityDepthData
|
||||
from .modality_lighting import ModalityLightingData
|
||||
from .point_of_view import PointOfViewData
|
||||
from .salience import SalienceData
|
||||
from .visual_syntax import VisualSyntaxData
|
||||
@@ -0,0 +1,11 @@
|
||||
"""Definition of Angle data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class AngleData(DataModel):
|
||||
"""Angle data model."""
|
||||
high: float
|
||||
eye_level: float
|
||||
low: float
|
||||
@@ -0,0 +1,11 @@
|
||||
"""Definition of ContactData data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class ContactData(DataModel):
|
||||
"""ContactData data model."""
|
||||
|
||||
offer: float
|
||||
demand: float
|
||||
@@ -0,0 +1,67 @@
|
||||
"""Definition of DataModel base class."""
|
||||
from __future__ import annotations
|
||||
|
||||
import random
|
||||
|
||||
from pydantic import BaseModel
|
||||
from pydantic import ValidationError
|
||||
|
||||
|
||||
class DataModel(BaseModel):
|
||||
"""DataModel base class."""
|
||||
|
||||
@classmethod
|
||||
def classname(cls) -> str:
|
||||
"""Return classname."""
|
||||
return cls.__name__
|
||||
|
||||
@classmethod
|
||||
def list_fields(cls) -> list[str]:
|
||||
"""List options that are stored as attributes."""
|
||||
return list(cls.model_fields.keys())
|
||||
|
||||
@classmethod
|
||||
def from_random(cls):
|
||||
"""Instantiate with random numbers."""
|
||||
kwargs = {field: random.random() for field in cls.list_fields()}
|
||||
return cls(**kwargs)
|
||||
|
||||
@classmethod
|
||||
def from_choice(cls, option: str):
|
||||
"""Instantiate from choice."""
|
||||
if option is None:
|
||||
raise ValidationError()
|
||||
assert isinstance(option, str), 'option is not a string'
|
||||
allowed_options_list = cls.list_fields()
|
||||
assert option in allowed_options_list, \
|
||||
f"{option} is not among allowed fields {allowed_options_list}"
|
||||
kwargs = {field: 0 for field in cls.list_fields()}
|
||||
kwargs[option] = 1
|
||||
return cls(**kwargs)
|
||||
|
||||
@classmethod
|
||||
def from_list(cls, data_list: list[float]):
|
||||
"""Instantiate from list of values."""
|
||||
kwargs = {key: val for key, val in zip(cls.list_fields(), data_list)}
|
||||
return cls(**kwargs)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
model_dict = self.model_dump()
|
||||
model_repr_str = f"{self.classname()}("
|
||||
model_repr_str += ', '.join([
|
||||
f"{field}={value:.3f}"
|
||||
for field, value
|
||||
in model_dict.items()
|
||||
])
|
||||
model_repr_str += ')'
|
||||
return model_repr_str
|
||||
|
||||
def highest_score_field(self) -> str:
|
||||
"""Return name of field with highest score."""
|
||||
model_dict = self.model_dump()
|
||||
return max(model_dict, key=lambda k: model_dict[k])
|
||||
|
||||
def highest_score_value(self) -> float:
|
||||
"""Return value of field with highest score."""
|
||||
model_dict = self.model_dump()
|
||||
return max(model_dict.values())
|
||||
@@ -0,0 +1,12 @@
|
||||
"""Definition of DistanceData data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class DistanceData(DataModel):
|
||||
"""DistanceData data model."""
|
||||
|
||||
long: float
|
||||
medium: float
|
||||
close: float
|
||||
@@ -0,0 +1,13 @@
|
||||
"""Definition of FramingData data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class FramingData(DataModel):
|
||||
"""FramingData data model."""
|
||||
|
||||
frame_lines: float
|
||||
empty_space: float
|
||||
colour_contrast: float
|
||||
form_contrast: float
|
||||
@@ -0,0 +1,12 @@
|
||||
"""Definition of InformationValueData data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class InformationValueData(DataModel):
|
||||
"""InformationValueData data model."""
|
||||
|
||||
given_new: float
|
||||
ideal_real: float
|
||||
central_marginal: float
|
||||
@@ -0,0 +1,12 @@
|
||||
"""Definition of ModalityColorData data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class ModalityColorData(DataModel):
|
||||
"""ModalityColorData data model."""
|
||||
|
||||
high: float
|
||||
medium: float
|
||||
low: float
|
||||
@@ -0,0 +1,12 @@
|
||||
"""Definition of ModalityDepthData data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class ModalityDepthData(DataModel):
|
||||
"""ModalityDepthData data model."""
|
||||
|
||||
high: float
|
||||
medium: float
|
||||
low: float
|
||||
@@ -0,0 +1,12 @@
|
||||
"""Definition of ModalityLightingData data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class ModalityLightingData(DataModel):
|
||||
"""ModalityLightingData data model."""
|
||||
|
||||
high: float
|
||||
medium: float
|
||||
low: float
|
||||
@@ -0,0 +1,11 @@
|
||||
"""Definition of PointOfViewData data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class PointOfViewData(DataModel):
|
||||
"""PointOfViewData data model."""
|
||||
|
||||
frontal: float
|
||||
oblique: float
|
||||
@@ -0,0 +1,14 @@
|
||||
"""Definition of SalienceData data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class SalienceData(DataModel):
|
||||
"""SalienceData data model."""
|
||||
|
||||
size: float
|
||||
colour: float
|
||||
tone: float
|
||||
form: float
|
||||
positioning: float
|
||||
@@ -0,0 +1,27 @@
|
||||
"""Definition of VisualSyntaxData data model."""
|
||||
from __future__ import annotations
|
||||
|
||||
from .data_model import DataModel
|
||||
|
||||
|
||||
class VisualSyntaxData(DataModel):
|
||||
"""VisualSyntaxData data model."""
|
||||
|
||||
non_transactional_action: float
|
||||
non_transactional_reaction: float
|
||||
unidirectional_transactional_action: float
|
||||
unidirectional_transactional_reaction: float
|
||||
bidirectional_transactional_action: float
|
||||
bidirectional_transactional_reaction: float
|
||||
conversion: float
|
||||
speech_process: float
|
||||
classification_overt_taxonomy: float
|
||||
analytical_exhaustive: float
|
||||
analytical_disarranged: float
|
||||
analytical_temporal: float
|
||||
analytical_distributed: float
|
||||
analytical_topological: float
|
||||
analytical_exploded: float
|
||||
analytical_inclusive: float
|
||||
symbolic_suggestive: float
|
||||
symbolic_attributive: float
|
||||
Reference in New Issue
Block a user