mirror of
https://github.com/microsoft/autogen.git
synced 2025-09-14 02:35:40 +00:00
a0b318b12e
* version update post release v1.2.2 * automl option * import pandas * remove automl.utils * default * test * type hint and version update * dependency update * link to open in colab * use packging.version to close #725 --------- Co-authored-by: Li Jiang <lijiang1@microsoft.com> Co-authored-by: Li Jiang <bnujli@gmail.com>
14 lines
428 B
Python
14 lines
428 B
Python
from typing import Optional, Union
|
|
import numpy as np
|
|
from flaml.automl.task.generic_task import GenericTask
|
|
from flaml.automl.task.task import Task
|
|
from flaml.automl.data import DataFrame, Series
|
|
|
|
|
|
def task_factory(
|
|
task_name: str,
|
|
X_train: Optional[Union[np.ndarray, DataFrame]] = None,
|
|
y_train: Optional[Union[np.ndarray, DataFrame, Series]] = None,
|
|
) -> Task:
|
|
return GenericTask(task_name, X_train, y_train)
|