forked from platypush/platypush
1f1fefca9d
TODO: Extend neural network implementation to work also with e.g. input from images, sounds or binary
72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
from typing import Optional, Dict, Union
|
|
|
|
from platypush.message.event import Event
|
|
|
|
|
|
class TensorflowEvent(Event):
|
|
def __init__(self, model: str, logs: Optional[Dict[str, Union[int, float]]], *args, **kwargs):
|
|
"""
|
|
:param model: Name of the Tensorflow model.
|
|
:param logs: Logs and metrics.
|
|
"""
|
|
super().__init__(*args, model=model, logs=logs, **kwargs)
|
|
|
|
|
|
class TensorflowEpochStartedEvent(TensorflowEvent):
|
|
"""
|
|
Triggered when a Tensorflow model training/evaluation epoch begins.
|
|
"""
|
|
def __init__(self, epoch: int, *args, **kwargs):
|
|
"""
|
|
:param epoch: Epoch index.
|
|
"""
|
|
super().__init__(*args, epoch=epoch, **kwargs)
|
|
|
|
|
|
class TensorflowEpochEndedEvent(TensorflowEvent):
|
|
"""
|
|
Triggered when a Tensorflow model training/evaluation epoch ends.
|
|
"""
|
|
def __init__(self, epoch: int, *args, **kwargs):
|
|
"""
|
|
:param epoch: Epoch index.
|
|
"""
|
|
super().__init__(*args, epoch=epoch, **kwargs)
|
|
|
|
|
|
class TensorflowBatchStartedEvent(TensorflowEvent):
|
|
"""
|
|
Triggered when a Tensorflow model training/evaluation batch starts being processed.
|
|
"""
|
|
def __init__(self, batch: int, *args, **kwargs):
|
|
"""
|
|
:param batch: Batch index.
|
|
"""
|
|
super().__init__(*args, batch=batch, **kwargs)
|
|
|
|
|
|
class TensorflowBatchEndedEvent(TensorflowEvent):
|
|
"""
|
|
Triggered when a the processing of a Tensorflow model training/evaluation batch ends.
|
|
"""
|
|
def __init__(self, batch: int, *args, **kwargs):
|
|
"""
|
|
:param batch: Batch index.
|
|
"""
|
|
super().__init__(*args, batch=batch, **kwargs)
|
|
|
|
|
|
class TensorflowTrainStartedEvent(TensorflowEvent):
|
|
"""
|
|
Triggered when a Tensorflow model starts being trained.
|
|
"""
|
|
|
|
|
|
class TensorflowTrainEndedEvent(TensorflowEvent):
|
|
"""
|
|
Triggered when the training phase of a Tensorflow model ends.
|
|
"""
|
|
|
|
|
|
# vim:sw=4:ts=4:et:
|