optima-ml 0.3.2a5.dev0__tar.gz → 0.3.2a5.dev1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (51) hide show
  1. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/builtin/evaluation.py +4 -1
  2. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/optima.py +1 -1
  3. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/PKG-INFO +2 -1
  4. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/optima_ml.egg-info/PKG-INFO +2 -1
  5. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/optima_ml.egg-info/requires.txt +1 -0
  6. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/setup.py +1 -0
  7. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/LICENSE +0 -0
  8. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/__init__.py +0 -0
  9. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/__main__.py +0 -0
  10. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/builtin/__init__.py +0 -0
  11. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/builtin/figures_of_merit.py +0 -0
  12. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/builtin/inputs.py +0 -0
  13. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/builtin/model.py +0 -0
  14. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/builtin/search_space.py +0 -0
  15. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/core/__init__.py +0 -0
  16. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/core/evaluation.py +0 -0
  17. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/core/inputs.py +0 -0
  18. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/core/model.py +0 -0
  19. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/core/search_space.py +0 -0
  20. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/core/tools.py +0 -0
  21. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/core/training.py +0 -0
  22. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/core/variable_optimization.py +0 -0
  23. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/defaults.py +0 -0
  24. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/hardware_configs/Dresden_Taurus.py +0 -0
  25. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/hardware_configs/__init__.py +0 -0
  26. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/hardware_configs/common.py +0 -0
  27. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/hardware_configs/helpers.py +0 -0
  28. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/helpers/__init__.py +0 -0
  29. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/helpers/extract_data_from_NTuples.py +0 -0
  30. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/helpers/manage_ray_nodes.py +0 -0
  31. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/keras/__init__.py +0 -0
  32. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/keras/model.py +0 -0
  33. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/keras/tools.py +0 -0
  34. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/keras/training.py +0 -0
  35. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/lightning/__init__.py +0 -0
  36. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/lightning/inputs.py +0 -0
  37. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/lightning/training.py +0 -0
  38. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/resources/__init__.py +0 -0
  39. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/resources/config_verification.py +0 -0
  40. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/OPTIMA/resources/pbt_with_seed.py +0 -0
  41. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/README.md +0 -0
  42. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/optima_ml.egg-info/SOURCES.txt +0 -0
  43. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/optima_ml.egg-info/dependency_links.txt +0 -0
  44. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/optima_ml.egg-info/entry_points.txt +0 -0
  45. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/optima_ml.egg-info/top_level.txt +0 -0
  46. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/setup.cfg +0 -0
  47. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/tests/test_builtin.py +0 -0
  48. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/tests/test_core.py +0 -0
  49. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/tests/test_integration.py +0 -0
  50. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/tests/test_integration_sameMachine.py +0 -0
  51. {optima_ml-0.3.2a5.dev0 → optima_ml-0.3.2a5.dev1}/tests/test_preprocessing.py +0 -0
@@ -578,7 +578,10 @@ def evaluate(
578
578
  # for each contribution to the histogram, we need to provide the type of contribution (step or scatter), the
579
579
  # index of the reference to use for the ratio subplot, and bin content itself..
580
580
  bin_contents_with_type = zip(["step"] * num_classes, [None] * num_classes, bin_contents)
581
- colors = sns.color_palette()
581
+ if (num_classes <= 10 and not explicit_testing_dataset) or num_classes <= 5:
582
+ colors = sns.color_palette()
583
+ else:
584
+ colors = sns.color_palette("husl", num_classes if not explicit_testing_dataset else 2 * num_classes)
582
585
  colors_errors = [None] * (num_classes - 1) + ["0.4"] # only the uppermost contribution should have error bars
583
586
  if class_labels is not None:
584
587
  legend_labels = class_labels if not binary_classification else class_labels[::-1]
@@ -2,7 +2,7 @@
2
2
 
3
3
  __author__ = "E. Bachmann"
4
4
  __licence__ = "GPL3"
5
- __version__ = "0.3.2alpha5_dev"
5
+ __version__ = "0.3.2alpha5_dev1"
6
6
  __maintainer__ = "E. Bachmann"
7
7
 
8
8
  import os
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: optima-ml
3
- Version: 0.3.2a5.dev0
3
+ Version: 0.3.2a5.dev1
4
4
  Summary: Distributed hyperparameter optimization and input variable selection for artificial neural networks.
5
5
  Home-page: https://gitlab.cern.ch/atlas-germany-dresden-vbs-group/optima
6
6
  Author: E. Bachmann
@@ -29,6 +29,7 @@ Requires-Dist: matplotlib
29
29
  Requires-Dist: seaborn
30
30
  Requires-Dist: tabulate
31
31
  Requires-Dist: scikit-learn
32
+ Requires-Dist: dill
32
33
  Provides-Extra: keras
33
34
  Requires-Dist: tensorflow==2.15.0; extra == "keras"
34
35
  Provides-Extra: lightning
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: optima-ml
3
- Version: 0.3.2a5.dev0
3
+ Version: 0.3.2a5.dev1
4
4
  Summary: Distributed hyperparameter optimization and input variable selection for artificial neural networks.
5
5
  Home-page: https://gitlab.cern.ch/atlas-germany-dresden-vbs-group/optima
6
6
  Author: E. Bachmann
@@ -29,6 +29,7 @@ Requires-Dist: matplotlib
29
29
  Requires-Dist: seaborn
30
30
  Requires-Dist: tabulate
31
31
  Requires-Dist: scikit-learn
32
+ Requires-Dist: dill
32
33
  Provides-Extra: keras
33
34
  Requires-Dist: tensorflow==2.15.0; extra == "keras"
34
35
  Provides-Extra: lightning
@@ -7,6 +7,7 @@ matplotlib
7
7
  seaborn
8
8
  tabulate
9
9
  scikit-learn
10
+ dill
10
11
 
11
12
  [keras]
12
13
  tensorflow==2.15.0
@@ -34,6 +34,7 @@ REQUIRED = [
34
34
  'seaborn',
35
35
  'tabulate',
36
36
  'scikit-learn',
37
+ 'dill'
37
38
  ] + (['multiprocess'] if sys.platform == 'darwin' else [])
38
39
 
39
40