From 7a62c92630aa49caa2c5f882d715c62cc03bc410 Mon Sep 17 00:00:00 2001 From: Kevin M Jablonka <32935233+kjappelbaum@users.noreply.github.com> Date: Sun, 1 Jan 2023 14:58:24 +0100 Subject: [PATCH] docs: fix typo in tabularregression docstring (#1503) --- flash/tabular/regression/model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/flash/tabular/regression/model.py b/flash/tabular/regression/model.py index 21ca00724c..0840c5cab1 100644 --- a/flash/tabular/regression/model.py +++ b/flash/tabular/regression/model.py @@ -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`