From c933798f7f00fe356245d0a6aa2435e0340472fd Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Sun, 6 Aug 2023 00:40:14 +0200
Subject: [PATCH] [doc] Remove traces of signs-to-tb "model"
---
doc/api.rst | 2 +-
doc/config.rst | 2 +-
doc/usage/training.rst | 5 +++--
3 files changed, 5 insertions(+), 4 deletions(-)
diff --git a/doc/api.rst b/doc/api.rst
index 41fc340f..5bfd2b2f 100644
--- a/doc/api.rst
+++ b/doc/api.rst
@@ -82,7 +82,7 @@ CNN and other models implemented.
ptbench.models.densenet
ptbench.models.normalizer
ptbench.models.logistic_regression
- ptbench.models.signs_to_tb
+ ptbench.models.mlp
.. _ptbench.api.engines:
diff --git a/doc/config.rst b/doc/config.rst
index 34025f7f..cedc8c07 100644
--- a/doc/config.rst
+++ b/doc/config.rst
@@ -26,8 +26,8 @@ Models
ptbench.models.config.densenet_pretrained
ptbench.models.config.densenet_rs
ptbench.models.config.logistic_regression
+ ptbench.models.config.mlp
ptbench.models.config.pasa
- ptbench.models.config.signs_to_tb
.. _ptbench.config.datamodules:
diff --git a/doc/usage/training.rst b/doc/usage/training.rst
index 67b3918c..5eb974cc 100644
--- a/doc/usage/training.rst
+++ b/doc/usage/training.rst
@@ -61,12 +61,13 @@ Montgomery dataset:
ptbench train -vv logistic_regression montgomery_rs --batch-size=4 --epochs=20
-To train Signs_to_TB using predictions from DensenetForRS on the Shenzhen
+To train an multi-layer perceptron (MLP) using predictions from a densenet
+pre-trained to detect radiological findings (using NIH CXR-14), on the Shenzhen
dataset:
.. code:: sh
- ptbench train -vv signs_to_tb shenzhen_rs --batch-size=4 --epochs=20
+ ptbench train -vv mlp shenzhen_rs --batch-size=4 --epochs=20
.. include:: ../links.rst
--
GitLab