.gitignore

@@ -3,4 +3,5 @@ __pycache__
 .idea/
 *.egg-info
 test.json
+.vscode
 

chillax/methods/sample_weight_by_ic.py

@@ -97,10 +97,10 @@ class SampleWeightByICSampleTransfomer(SampleTransformer, instrumentation.Observ
 
         # Apply softmax
         sample_weights = {
-            sample_uid: sample_ic_exp_tf / ic_exp_tf_sum
+            sample_uid: (sample_ic_exp_tf * len(samples)) / ic_exp_tf_sum
             for sample_uid, sample_ic_exp_tf in sample_ics_exp_tf.items()
         }
-        sample_weight_array = np.asarray(list(sample_weights.values())) * len(samples)
+        sample_weight_array = np.asarray(list(sample_weights.values()))
         self.log_info(
             f"Weight*cnt mean {np.mean(sample_weight_array)}, std {np.std(sample_weight_array)}, median {np.median(sample_weight_array)}"
         )

chillax/version.py

@@ -1 +1 @@
-__version__ = "0.1a14"
+__version__ = "0.1a15"

tests/test_experiment.py

@@ -14,3 +14,7 @@ def test_experiment():
     example_config_files = ["examples/configuration.json"]
 
     experiment_selfsupervised.main(example_config_files)
+
+
+if __name__ == "__main__":
+    test_experiment()