rockfish.actions.transformer
rockfish.actions.transformer
Classes
GenerateTabTransformerConfig
Configuration for the GenerateTabTransformer action.
Attributes:
| Name | Type | Description |
|---|---|---|
records |
Optional[int]
|
Number of records to generate. If None, the number of records in the training set up to the maximum of 1000. |
GenerateTimeTransformerConfig
Configuration for the GenerateTimeTransformer action.
Attributes:
| Name | Type | Description |
|---|---|---|
sessions |
Optional[int]
|
Number of sessions to generate. If None, the number of sessions in the training set up to the maximum of 1000. |
TrainTabTransformerConfig
Configuration for the TrainTabTransformer action.
Attributes:
| Name | Type | Description |
|---|---|---|
rtf |
TrainTabConfig
|
|
encoder |
DatasetConfig
|
|
model_labels |
LabelDict
|
Additional labels to add to the Model. |
TrainTabConfig
TrainTimeTransformerConfig
Configuration for the TrainTimeTransformer action.
Attributes:
| Name | Type | Description |
|---|---|---|
rtf |
TrainTimeConfig
|
|
encoder |
DatasetConfig
|
|
model_labels |
LabelDict
|
Additional labels to add to the Model. |
TrainTimeConfig
Attributes:
| Name | Type | Description |
|---|---|---|
num_bootstrap |
int
|
|
parent |
ParentConfig
|
Configuration for the parent/metadata table. |
child |
ChildConfig
|
Configuration for the child/measurement table. |
ParentConfig
Attributes:
| Name | Type | Description |
|---|---|---|
epochs |
int
|
Number of epochs to train for. |
batch_size |
int
|
|
gpt2_config |
GPT2Config
|
|