datarobot_genai.drtools.predictive.training

training

Tools for analyzing datasets and suggesting ML use cases.

UseCaseSuggestion dataclass

Represents a suggested use case based on dataset analysis.

Source code in datarobot_genai/drtools/predictive/training.py

@dataclass
class UseCaseSuggestion:
    """Represents a suggested use case based on dataset analysis."""

    name: str
    description: str
    suggested_target: str
    problem_type: str
    confidence: float
    reasoning: str

DatasetInsight dataclass

Contains insights about a dataset for use case discovery.

Source code in datarobot_genai/drtools/predictive/training.py

@dataclass
class DatasetInsight:
    """Contains insights about a dataset for use case discovery."""

    total_columns: int
    total_rows: int
    numerical_columns: list[str]
    categorical_columns: list[str]
    datetime_columns: list[str]
    text_columns: list[str]
    potential_targets: list[str]
    missing_data_summary: dict[str, float]