Models
class TabularModel
TabularModel.build
Tabular model to generate synthetic tabular relational data.
Arguments:
preproc
- ATabularPreproc
object.size
- The size configuration of the model. Could be either aSize
object, aTabularModelSize
object or a string representation of such objects.dropout
- The dropout probability.block_size
- Maximum sequence length that the model can process. If None, no maximum sequence length is imposed.
Returns:
A TabularModel
instance.
TabularModel.generate
Generate synthetic relational data.
Arguments:
n_samples
- Desired number of samples in the root table.batch_size
- Batch size used during generation. If 0, generate all data in a single batch.temp
- Temperature parameter for sampling.
Returns:
A RelationalData
object.
class TextModel
TextModel.build
Text model to generate synthetic text columns of a table which is part of a relational structure.
Arguments:
preproc
- ATextPreproc
object.size
- The size configuration of the model. Could be either aSize
object, aTextModelSize
object or a string representation of such objects.block_size
- Maximum text sequence length that the model can process.dropout
- The dropout probability.
Returns:
A TextModel
instance.
TextModel.generate
Generate text columns in the current table.
Arguments:
data
- ARelationalData
object containing synthetic data.batch_size
- Batch size used during generation. If 0, generate all data in a single batch.max_text_len
- Maximum length for the generated text. If 0, the maximum possible value is used.temp
- Temperature parameter for sampling.
Returns:
A RelationalData
object.
class Size
Enumeration class representing different model sizes. Supported sizes are: SMALL, MEDIUM and LARGE.
class TabularModelSize
Model size for TabularModel
objects.
Arguments:
n_layers
- Number of internal layers.h
- Number of heads.d
- Size of the internal dimension.
TabularModelSize.from_size
Create an instance based on a given Size
or its string representation.
Arguments:
size
- ASize
object.
class TextModelSize
Model size for TextModel
objects.
Arguments:
n_layers
- Number of internal layers.h
- Number of heads.d
- Size of the internal dimension.
TextModelSize.from_size
Create an instance based on a given Size
or its string representation.
Arguments:
size
- ASize
or a str representing aSize
.