From 431244fe25d2125cdccb0a3233b84a4f11b04d12 Mon Sep 17 00:00:00 2001 From: Kevin M Jablonka <32935233+kjappelbaum@users.noreply.github.com> Date: Thu, 29 Dec 2022 08:18:45 +0100 Subject: [PATCH] docs: fix typo in tabularregression docstring --- 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`