mirror of
https://github.com/microsoft/autogen.git
synced 2025-09-14 10:45:54 +00:00
16 lines
411 B
Python
16 lines
411 B
Python
|
from typing import Optional, Union
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
import pandas as pd
|
||
|
|
|
||
|
|
from flaml.automl.task.generic_task import GenericTask
|
||
|
|
from flaml.automl.task.task import Task
|
||
|
|
|
||
|
|
|
||
|
|
def task_factory(
|
||
|
|
task_name: str,
|
||
|
|
X_train: Optional[Union[np.ndarray, pd.DataFrame]] = None,
|
||
|
|
y_train: Optional[Union[np.ndarray, pd.DataFrame, pd.Series]] = None,
|
||
|
|
) -> Task:
|
||
|
|
return GenericTask(task_name, X_train, y_train)
|