Move files
This commit is contained in:
parent
d2af1b8151
commit
f160b07163
159 changed files with 104 additions and 88 deletions
3
src/great_ai/output_views/__init__.py
Normal file
3
src/great_ai/output_views/__init__.py
Normal file
|
|
@ -0,0 +1,3 @@
|
|||
from .classification_output import ClassificationOutput
|
||||
from .multi_label_classification_output import MultiLabelClassificationOutput
|
||||
from .regression_output import RegressionOutput
|
||||
9
src/great_ai/output_views/classification_output.py
Normal file
9
src/great_ai/output_views/classification_output.py
Normal file
|
|
@ -0,0 +1,9 @@
|
|||
from typing import Any, Optional, Union
|
||||
|
||||
from ..helper import HashableBaseModel
|
||||
|
||||
|
||||
class ClassificationOutput(HashableBaseModel):
|
||||
label: Union[str, int]
|
||||
confidence: float
|
||||
explanation: Optional[Any]
|
||||
|
|
@ -0,0 +1,8 @@
|
|||
from typing import List
|
||||
|
||||
from ..helper import HashableBaseModel
|
||||
from .classification_output import ClassificationOutput
|
||||
|
||||
|
||||
class MultiLabelClassificationOutput(HashableBaseModel):
|
||||
labels: List[ClassificationOutput] = []
|
||||
8
src/great_ai/output_views/regression_output.py
Normal file
8
src/great_ai/output_views/regression_output.py
Normal file
|
|
@ -0,0 +1,8 @@
|
|||
from typing import Any, Optional, Union
|
||||
|
||||
from ..helper import HashableBaseModel
|
||||
|
||||
|
||||
class RegressionOutput(HashableBaseModel):
|
||||
value: Union[int, float]
|
||||
explanation: Optional[Any]
|
||||
1
src/great_ai/output_views/sequence_labeling_output.py
Normal file
1
src/great_ai/output_views/sequence_labeling_output.py
Normal file
|
|
@ -0,0 +1 @@
|
|||
# todo
|
||||
Loading…
Add table
Add a link
Reference in a new issue