Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

docs: fix typo in tabularregression docstring #1503

Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions flash/tabular/regression/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,10 +40,10 @@ class TabularRegressor(RegressionAdapterTask):
parameters: The parameters computed from the training data (can be obtained from the ``parameters`` attribute of
the ``TabularRegressionData`` object containing your training data).
embedding_sizes: List of (num_classes, emb_dim) to form categorical embeddings.
cat_dims: Number of distinct values for each categorical column
num_features: Number of columns in table
backbone: name of the model to use
loss_fn: Loss function for training, defaults to cross entropy.
cat_dims: Number of distinct values for each categorical column.
num_features: Number of columns in table.
backbone: name of the model to use.
loss_fn: Loss function for training, defaults to mean squared error.
optimizer: Optimizer to use for training.
lr_scheduler: The LR scheduler to use during training.
metrics: Metrics to compute for training and evaluation. Can either be an metric from the `torchmetrics`
Expand Down