Add docs for LLMAttribution (#1217)

aobo-y · facebook-github-bot · commit 645af9691a54 · 2023-12-04T07:06:01.000-08:00
Summary: as title Pull Request resolved: #1217 Reviewed By: vivekmig Differential Revision: D51803181 Pulled By: aobo-y fbshipit-source-id: 81d9bb4bb5d839ac12b98e2344c87ec59b5f0718
diff --git a/sphinx/source/index.rst b/sphinx/source/index.rst
@@ -12,6 +12,7 @@ Captum API Reference
    :caption: API Reference
 
    attribution
+   llm_attr
    noise_tunnel
    layer
    neuron
diff --git a/sphinx/source/llm_attr.rst b/sphinx/source/llm_attr.rst
@@ -0,0 +1,21 @@
+LLM Attribution Classes
+========================
+
+LLMAttribution
+^^^^^^^^^^^^^^^^^^^^^^^^
+
+.. autoclass:: captum.attr.LLMAttribution
+    :members:
+
+LLMGradientAttribution
+^^^^^^^^^^^^^^^^^^^^^^^^
+
+.. autoclass:: captum.attr.LLMGradientAttribution
+    :members:
+
+
+LLMAttributionResult
+^^^^^^^^^^^^^^^^^^^^^^^^
+
+.. autoclass:: captum.attr.LLMAttributionResult
+    :members:
diff --git a/sphinx/source/utilities.rst b/sphinx/source/utilities.rst
@@ -1,6 +1,18 @@
 Utilities
 ==========
 
+Interpretable Input
+^^^^^^^^^^^^^^^^^^^^^
+.. autoclass:: captum.attr.InterpretableInput
+    :members:
+
+.. autoclass:: captum.attr.TextTemplateInput
+    :members:
+
+.. autoclass:: captum.attr.TextTokenInput
+    :members:
+
+
 Visualization
 ^^^^^^^^^^^^^^
 
@@ -48,3 +60,9 @@ Linear Models
     :members:
 .. autoclass:: captum._utils.models.linear_model.SGDRidge
     :members:
+
+Baselines
+^^^^^^^^^^^^^^^^
+
+.. autoclass:: captum.attr.ProductBaselines
+    :members:
