Fixed finetuning docs

datamol-io · Dec 21, 2023 · 7fe8c39 · 7fe8c39
1 parent 09e0d6f
commit 7fe8c39
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/docs/api/graphium.finetuning.md b/docs/api/graphium.finetuning.md
@@ -1,12 +1,13 @@
 graphium.finetuning
 ====================
-module for finetuning models
+Module for finetuning models and doing linear probing (fingerprinting).
+
+::: graphium.finetuning.finetuning.GraphFinetuning
+
+::: graphium.finetuning.finetuning_architecture.FullGraphFinetuningNetwork
+
+::: graphium.finetuning.finetuning_architecture.PretrainedModel
+
+::: graphium.finetuning.finetuning_architecture.FinetuningHead
 
 ::: graphium.finetuning.fingerprinting.Fingerprinter
-    <!-- options:
-        filters: ["!^_"]
-        separate_signature: true
-        show_signature_annotations: true
-        line_length: 80
-        merge_init_into_class: true
-        members_order: source -->