Merge pull request #21 from Franblueee:vaeabmil

Adding VAEABMIL

Author: Franblueee

docs/api/datasets/tadmil_dataset.md

@@ -0,0 +1,7 @@
+# Traffic Anomaly Detection (TAD) MIL Dataset
+
+::: torchmil.datasets.TADMILDataset
+    options:
+        members:
+            - __init__
+            - __getitem__

docs/api/datasets/video_classification_dataset.md

@@ -0,0 +1,7 @@
+# Video Classification Dataset
+
+::: torchmil.datasets.VideoClassificationDataset
+    options:
+        members:
+            - __init__
+            - __getitem__

docs/api/index.md

@@ -46,6 +46,8 @@
 - [Sm operator](nn/sm.md)
 - [Max Pool](nn/max_pool.md)
 - [Mean Pool](nn/mean_pool.md)
+- [Multi Layer Perceptron](nn/mlp.md)
+- [Variational Autoencoder](nn/variational_autoencoder.md)
 
 ## Models: [torchmil.models](models/index.md)
 - [Introduction](models/index.md)
@@ -66,6 +68,7 @@
 - [TransformerProbSmoothABMIL](models/transformer_prob_smooth_abmil.md)
 - [TransMIL](models/transmil.md)
 - [SETMIL](models/setmil.md)
+- [VAEABMIL](models/vaeabmil.md)
 
 ## Visualize: [torchmil.visualize](visualize/index.md)
 - [Introduction](visualize/index.md)

docs/api/models/index.md

@@ -19,3 +19,4 @@ They all inherit from the [General MIL model](mil_model.md) class, which provide
 - [TransMIL](transmil.md)
 - [SETMIL](setmil.md)
 - [IIBMIL](iibmil.md)
+- [VAEABMIL](vaeabmil.md)

docs/api/models/vaeabmil.md

@@ -0,0 +1,10 @@
+# VAEABMIL
+
+::: torchmil.models.VAEABMIL
+    options:
+        members:
+            - __init__
+            - forward
+            - compute_loss
+            - predict
+            - log_marginal_likelihood_importance_sampling

docs/api/nn/index.md

@@ -27,3 +27,5 @@ These modules are designed to be flexible and easy to use, allowing you to build
 - [Sm operator](sm.md)
 - [Max Pool](max_pool.md)
 - [Mean Pool](mean_pool.md)
+- [Multi Layer Perceptron](mlp.md)
+- [Variational Autoencoder](variational_autoencoder.md)

docs/api/nn/mlp.md

@@ -0,0 +1,6 @@
+# MLP
+::: torchmil.nn.MLP
+    options:
+        members:
+        - __init__
+        - forward

docs/api/nn/variational_autoencoder.md

@@ -0,0 +1,24 @@
+# Variational Autoencoder
+::: torchmil.nn.VariationalAutoEncoder
+    options:
+        members:
+        - __init__
+        - get_reparameterized_samples
+        - get_raw_output_enc
+        - get_raw_output_dec
+        - forward
+        - get_posterior_samples
+        - complete_forward_samples
+        - compute_loss
+        - _kl_prior
+        - _diagonal_log_gaussian_pdf
+        - log_marginal_likelihood_importance_sampling
+-------------------------
+::: torchmil.nn.VariationalAutoEncoderMIL
+    options:
+        members:
+        - __init__
+        - forward
+        - log_marginal_likelihood_importance_sampling
+        - compute_loss
+        - complete_forward_samples

mkdocs.yml

@@ -136,9 +136,11 @@ nav:
       - api/datasets/wsi_dataset.md
       - api/datasets/trident_wsi_dataset.md
       - api/datasets/binary_classification_dataset.md
+      - api/datasets/video_classification_dataset.md
       - api/datasets/camelyon16mil_dataset.md
       - api/datasets/pandamil_dataset.md
       - api/datasets/rsnamil_dataset.md
+      - api/datasets/tadmil_dataset.md
       - api/datasets/false_frequency_dataset.md
       - api/datasets/mc_standard_dataset.md
       - api/datasets/sc_standard_dataset.md
@@ -171,6 +173,8 @@ nav:
       - api/nn/sm.md
       - api/nn/mean_pool.md
       - api/nn/max_pool.md
+      - api/nn/mlp.md
+      - api/nn/variational_autoencoder.md
     - torchmil.models:
       - api/models/index.md
       - api/models/mil_model.md
@@ -191,6 +195,7 @@ nav:
       - api/models/transmil.md
       - api/models/setmil.md
       - api/models/iibmil.md
+      - api/models/vaeabmil.md
     - torchmil.visualize:
       - api/visualize/index.md
       - api/visualize/vis_ctscan.md