mirror of
https://github.com/NicolasBohn/NexQuant.git
synced 2026-07-28 07:57:44 +00:00
Xinjie Shen
f20dc4482b
* upload example code for model implementation * Refactor Model Implement * add export --------- Co-authored-by: Young <afe.young@gmail.com>
33 lines
850 B
Python
33 lines
850 B
Python
from abc import ABC, abstractmethod
|
|
from typing import List, Sequence
|
|
|
|
from rdagent.core.task import (
|
|
BaseTask,
|
|
TaskImplementation,
|
|
)
|
|
|
|
class TaskGenerator(ABC):
|
|
@abstractmethod
|
|
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.
|
|
|
|
"""
|
|
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]
|
|
|
|
"""
|
|
|
|
|