2024-06-14 12:59:44 +08:00
|
|
|
from abc import ABC, abstractmethod
|
2024-06-21 16:41:34 +08:00
|
|
|
from typing import List, Sequence
|
2024-06-14 12:59:44 +08:00
|
|
|
|
|
|
|
|
from rdagent.core.task import (
|
2024-06-21 16:41:34 +08:00
|
|
|
BaseTask,
|
2024-06-14 12:59:44 +08:00
|
|
|
TaskImplementation,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
class TaskGenerator(ABC):
|
|
|
|
|
@abstractmethod
|
2024-06-21 16:41:34 +08:00
|
|
|
def generate(self, task_l: Sequence[BaseTask]) -> Sequence[TaskImplementation]:
|
|
|
|
|
"""
|
|
|
|
|
Task Generator should take in a sequence of tasks.
|
|
|
|
|
|
|
|
|
|
Because the schedule of different tasks is crucial for the final performance
|
|
|
|
|
due to it affects the learning process.
|
|
|
|
|
|
|
|
|
|
"""
|
2024-06-14 12:59:44 +08:00
|
|
|
raise NotImplementedError("generate method is not implemented.")
|
|
|
|
|
|
|
|
|
|
def collect_feedback(self, feedback_obj_l: List[object]):
|
|
|
|
|
"""
|
|
|
|
|
When online evaluation.
|
|
|
|
|
The preivous feedbacks will be collected to support advanced factor generator
|
|
|
|
|
|
|
|
|
|
Parameters
|
|
|
|
|
----------
|
|
|
|
|
feedback_obj_l : List[object]
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
|