From e310ddfd4a768b8294a52934c35e81d7b1656d25 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Danie=CC=88l=20de=20Kok?= Date: Fri, 8 Dec 2023 12:03:18 +0100 Subject: [PATCH] Add an infobox to the `TextCatReduce` section with an `TextCatCNN` anchor --- website/docs/api/architectures.mdx | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/website/docs/api/architectures.mdx b/website/docs/api/architectures.mdx index 643d66140..1e5f50512 100644 --- a/website/docs/api/architectures.mdx +++ b/website/docs/api/architectures.mdx @@ -1084,6 +1084,14 @@ A classifier that pools token hidden representations of each `Doc` using first, max or mean reduction and then applies a classification layer. Reductions are concatenated when multiple reductions are used. + + +`TextCatReduce` is a generalization of the older +[`TextCatCNN`](/api/legacy#TextCatCNN_v2) model. `TextCatCNN` always uses a mean +reduction, whereas `TextCatReduce` also supports first/max reductions. + + + | Name | Description | | ------------------- | ---------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | | `exclusive_classes` | Whether or not categories are mutually exclusive. ~~bool~~ |