Required Feature Group Types

To train a model under this use case, you will need to create feature groups of the following type(s):

Feature Group Type API Configuration Name Required Description
Finetuning Data FINETUNING_DATA True This dataset includes examples of (prompt, response) pairs to finetune the LLM.

Note: Once you upload the datasets under each Feature Group Type that comply with their respective required schemas, you will need to create Machine learning (ML) features that would be used to train your ML model(s). We use the term, "Feature Group" for a group of ML features (dataset columns) under a specific Feature Group Type. Our system support extensible schemas that enables you to provide any number of additional columns/features that you think are relevant to that Feature Group Type.


Feature Group: Finetuning Data

This dataset includes examples of (prompt, response) pairs to finetune the LLM.

Feature Mapping Feature Type Required Description
PROMPT Y Example prompt values
COMPLETION Y Example completion values