updates

.gitignore

@@ -155,3 +155,4 @@ run.py
 tests/figs
 tests/res
 tests/test.py
+/decision-fusion-framework/

docs/source/install.rst

@@ -41,7 +41,7 @@ For `MacOS` or `Unix` user:
 
     python3 -m build
 
-Once successfully executed, two files are generated in the `dist` subfolder within the project's root folder.
+Once successfully executed, two files are generated in the `dist` subfolder within the project's root folder. The `tar.gz` file is the source distribution and the `whl` file is the built distribution.
 
 Installation
 ------------

pyproject.toml

@@ -11,7 +11,7 @@ authors = [
 ]
 description = "Pusion (Python Universal Fusion) is a generic and automated framework for decision fusion written in Python."
 readme = "README.md"
-license={file = "LICENSE.txt"}
+license={file = "LICENSE"}
 requires-python = ">=3.6"
 classifiers = [
     "Programming Language :: Python :: 3",
@@ -24,7 +24,8 @@ dependencies = [
     "scikit-learn~=0.24.1",
     "setuptools~=54.2.0",
     "pandas~=1.2.3",
-    "matplotlib~=3.4.1"
+    "matplotlib~=3.4.1",
+    "torchmetrics~=0.7.2"
 ]
 keywords = ["decision fusion", "combining classifiers", "fusion", "pusion"]
 

src/pusion/__init__.py

@@ -21,7 +21,7 @@
 from pusion.util import *
 from pusion.util.constants import *
 from pusion.util.exceptions import *
-
+from pusion.util.generator import *
 
 # Maintain this static attributes according to implemented methods for ease of framework usage,
 # e.g., `pusion.Method.AutoCombiner`.

src/pusion/evaluation/evaluation_metrics.py

@@ -1,5 +1,5 @@
 import numpy as np
-import torch
+#import torch
 import torchmetrics
 from sklearn.metrics import *
 

tests/user_test_1.py

@@ -93,5 +93,8 @@ def main():
     print("------------- Runtimes ---------------")
     print("Total train time:", t_elapsed_train)
     print("Total combine time:", t_elapsed_combine)
-    print()
+
+
+if __name__ == '__main__':
+    main()