diff --git a/doc/api.rst b/doc/api.rst index 41fc340f311746d5585e6448126fbfbafe853899..5bfd2b2ff89183687536b4a65a85f8ef0591588d 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 34025f7f70d8a05299c67491c1115c06674c67ff..cedc8c0760a99467952171b791f17f06c39c0366 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 67b3918cee7640bdee8f0277ec5390142163c21e..5eb974cc9a0e0723f9e9e72ba4c183b0c582daf2 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