Bring-Your-Own-Task (BYOT)

The Bring-Your-Own-Task (BYOT) feature allows users to create custom tasks for training, evaluating, and deploying heart-related AI models. This feature is useful for creating custom workflows for a given application with minimal coding.

For an in-depth guide check out the BYOT Notebook Guide.

How it Works

Create a Task: Define a new task that inherits from the HKTask base class and implements the required methods: train, evaluate, export, and demo.

import heartkit as hk

class CustomTask(hk.HKTask):

    @staticmethod
    def train(params: hk.HKTrainParams) -> None:
        pass

    @staticmethod
    def evaluate(params: hk.HKTestParams) -> None:
        pass

    @staticmethod
    def export(params: hk.HKExportParams) -> None:
        pass

    @staticmethod
    def demo(params: hk.HKDemoParams) -> None:
        pass

Register the Task: Register the new task with the TaskFactory by calling the register method. This method takes the task name and the task class as arguments.
Python example
```
# Register the custom task
hk.TaskFactory.register("custom", CustomTask)
```
Use the Task: The new task can now be used with the TaskFactory to perform various operations such as training, evaluating, and deploying models.
Python example
```
# Define task parameters
params = hk.HKTrainParams(...)

# Get the custom task
task = hk.TaskFactory.get("custom")

# Train the model
task.train(params)
```