Model mixin
----------------------------------------------

Utilities for fit, inference, and evaluation of classifiers.

.. autoclass:: simba.mixins.train_model_mixin.TrainModelMixin
   :members:
   :undoc-members:

Batch random forest inference
----------------------------------------------

.. autoclass:: simba.model.inference_batch.InferenceBatch
   :members:
   :undoc-members:

Batch multi-animal random forest inference
----------------------------------------------

.. autoclass:: simba.model.inference_multi_animal_batch.InferenceMultiAnimalBatch
   :members:
   :undoc-members:

Batch multi-class random forest inference
----------------------------------------------

.. autoclass:: simba.model.inference_multiclass_batch.InferenceMulticlassBatch
   :members:
   :undoc-members:

Grid-search random forest classifiers
----------------------------------------------

.. autoclass:: simba.model.grid_search_rf.GridSearchRandomForestClassifier
   :members:
   :undoc-members:

Grid-search random forest multi-classifiers
----------------------------------------------

.. autoclass:: simba.model.grid_search_multiclass_rf.GridSearchMulticlassRandomForestClassifier
   :members:
   :undoc-members:

Random forest inference - validation
----------------------------------------------

.. autoclass:: simba.model.inference_validation.InferenceValidation
   :members:
   :undoc-members:

Fit random forest classifier
----------------------------------------------

.. autoclass:: simba.model.train_rf.TrainRandomForestClassifier
   :members:
   :undoc-members:


Fit random forest classifier - multi-class
----------------------------------------------

.. autoclass:: simba.model.train_multiclass_rf.TrainMultiClassRandomForestClassifier
   :members:
   :undoc-members:


Ordinal classifier methods
----------------------------------------------

.. autoclass:: simba.model.ordinal_clf.OrdinalClassifier
   :members:
   :undoc-members:


Regression - metrics
-------------------------------------------------

.. automodule:: simba.model.regression.metrics
   :members:
   :show-inheritance:

Regression - fit and transform
-------------------------------------------------

.. automodule:: simba.model.regression.model
   :members:
   :show-inheritance:

SAM2 segmentation inference
-------------------------------------------------

.. autoclass:: simba.model.sam_inference.SamInference
   :members:
   :undoc-members:

Fit YOLO model
-------------------------------------------------

.. autoclass:: simba.model.yolo_fit.FitYolo
   :members:
   :undoc-members:

YOLO bounding-box inference
-------------------------------------------------

.. autoclass:: simba.model.yolo_inference.YoloInference
   :members:
   :undoc-members: