Fix API docs rendering (#1041)

Nice, thanks!
jmschrei · Aug 25, 2023 · ee3c177 · ee3c177
1 parent 02b8b30
commit ee3c177
Showing 1 changed file with 13 additions and 21 deletions.
diff --git a/docs/api.rst b/docs/api.rst
@@ -62,33 +62,25 @@ This returns the matrix of log posterior probabilities for numerical stability.
 API Reference
 -------------
 
+Distributions
+=============
+
 .. automodule:: pomegranate.distributions
    :members: Bernoulli, Categorical, ConditionalCategorical, JointCategorical, DiracDelta, Exponential, Gamma, Normal, Poisson, StudentT, Uniform, ZeroInflated
 
-.. automodule:: pomegranate.bayes_classifier.BayesClassifier
-	:members:
-	:inherited-members:
+Models
+======
+
+.. autoclass:: pomegranate.bayes_classifier.BayesClassifier
 
-.. automodule:: pomegranate.gmm.GeneralMixtureModel
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.gmm.GeneralMixtureModel
 
-.. automodule:: pomegranate.hmm.DenseHMM
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.hmm.DenseHMM
 
-.. automodule:: pomegranate.hmm.SparseHMM
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.hmm.SparseHMM
 
-.. automodule:: pomegranate.markov_chain.MarkovChain
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.markov_chain.MarkovChain
 
-.. automodule:: pomegranate.bayesian_network.BayesianNetwork
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.bayesian_network.BayesianNetwork
 
-.. automodule:: pomegranate.factor_graph.FactorGraph
-	:members:
-	:inherited-members:
+.. autoclass:: pomegranate.factor_graph.FactorGraph