Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We鈥檒l occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEATURE] Add ability to specify dataset problem type during data import #690

Open
MartinBarus opened this issue Apr 26, 2024 · 1 comment
Labels
type/feature Feature request

Comments

@MartinBarus
Copy link

馃殌 Feature

LLM studio supports multiple problem types (Causal Modeling/ Classification), however, during data import, it expects dataset to be named exactly as the problem type, otherwise proper problem type can not be set.

Motivation

When importing custom dataset named train.csv for classification, it would be great to have the ability to select proper problem type.

@MartinBarus MartinBarus added the type/feature Feature request label Apr 26, 2024
@MartinBarus
Copy link
Author

Maybe add this snippet

items += [
                ui.dropdown(
                    name="dataset/import/cfg_file",
                    label="Problem Type",
                    required=True,
                    choices=choices_problem_types,
                    value=q.client["dataset/import/cfg_file"],
                    trigger=True,
                    tooltip=tooltips["experiments_problem_type"],
                )
            ]

here :https://github.com/h2oai/h2o-llmstudio/blob/d1acb59/llm_studio/app_utils/sections/dataset.py#L491

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
type/feature Feature request
Projects
None yet
Development

No branches or pull requests

1 participant