Spaces:
Sleeping
Sleeping
File size: 2,978 Bytes
33d4721 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
from typing import List, Optional, Union
from pydantic import Field
from autotrain.trainers.common import AutoTrainParams
class TabularParams(AutoTrainParams):
"""
TabularParams is a configuration class for tabular data training parameters.
Attributes:
data_path (str): Path to the dataset.
model (str): Name of the model to use. Default is "xgboost".
username (Optional[str]): Hugging Face Username.
seed (int): Random seed for reproducibility. Default is 42.
train_split (str): Name of the training data split. Default is "train".
valid_split (Optional[str]): Name of the validation data split.
project_name (str): Name of the output directory. Default is "project-name".
token (Optional[str]): Hub Token for authentication.
push_to_hub (bool): Whether to push the model to the hub. Default is False.
id_column (str): Name of the ID column. Default is "id".
target_columns (Union[List[str], str]): Target column(s) in the dataset. Default is ["target"].
categorical_columns (Optional[List[str]]): List of categorical columns.
numerical_columns (Optional[List[str]]): List of numerical columns.
task (str): Type of task (e.g., "classification"). Default is "classification".
num_trials (int): Number of trials for hyperparameter optimization. Default is 10.
time_limit (int): Time limit for training in seconds. Default is 600.
categorical_imputer (Optional[str]): Imputer strategy for categorical columns.
numerical_imputer (Optional[str]): Imputer strategy for numerical columns.
numeric_scaler (Optional[str]): Scaler strategy for numerical columns.
"""
data_path: str = Field(None, title="Data path")
model: str = Field("xgboost", title="Model name")
username: Optional[str] = Field(None, title="Hugging Face Username")
seed: int = Field(42, title="Seed")
train_split: str = Field("train", title="Train split")
valid_split: Optional[str] = Field(None, title="Validation split")
project_name: str = Field("project-name", title="Output directory")
token: Optional[str] = Field(None, title="Hub Token")
push_to_hub: bool = Field(False, title="Push to hub")
id_column: str = Field("id", title="ID column")
target_columns: Union[List[str], str] = Field(["target"], title="Target column(s)")
categorical_columns: Optional[List[str]] = Field(None, title="Categorical columns")
numerical_columns: Optional[List[str]] = Field(None, title="Numerical columns")
task: str = Field("classification", title="Task")
num_trials: int = Field(10, title="Number of trials")
time_limit: int = Field(600, title="Time limit")
categorical_imputer: Optional[str] = Field(None, title="Categorical imputer")
numerical_imputer: Optional[str] = Field(None, title="Numerical imputer")
numeric_scaler: Optional[str] = Field(None, title="Numeric scaler")
|