Skip to content

deploy

deploy(environment, model, input_transforms=None, output_transforms=None)

Deploy a trained model to a custom environment.

Parameters:

Name Type Description Default
environment ActiveEnvironment | IncrementalEnvironment

custom system environment

required
model ModelWithTransform

the trained model

required
input_transforms Transform | None

system specific transforms for model input

None
output_transforms Transform | None

system specific transforms for system input

None
Source code in src/flowcean/core/strategies/deploy.py
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
def deploy(
    environment: ActiveEnvironment | IncrementalEnvironment,
    model: ModelWithTransform,
    input_transforms: Transform | None = None,
    output_transforms: Transform | None = None,
) -> None:
    """Deploy a trained model to a custom environment.

    Args:
        environment: custom system environment
        model: the trained model
        input_transforms: system specific transforms for model input
        output_transforms: system specific transforms for system input
    """
    if input_transforms is None:
        input_transforms = Identity()
    if output_transforms is None:
        output_transforms = Identity()

    observation = environment.observe()
    output = model.predict(input_transforms.apply(observation))

    if isinstance(environment, ActiveEnvironment):
        environment.act(output_transforms.apply(output).collect())