Clemens-Alexander Brust 5 лет назад
Родитель
Сommit
65dd784085
1 измененных файлов с 8 добавлено и 3 удалено
  1. 8 3
      chillax/experiment_selfsupervised.py

+ 8 - 3
chillax/experiment_selfsupervised.py

@@ -13,6 +13,11 @@ def main():
     config = {
     config = {
         "evaluators": [{"name": "accuracy"}],
         "evaluators": [{"name": "accuracy"}],
         "with_wordnet": True,
         "with_wordnet": True,
+        "interactor": {
+            "name": "noisy_oracle",
+            "noise_model": "Deng2014",
+            "relabel_fraction": 0.95
+        },
         "dataset": {
         "dataset": {
             "name": "nabirds",
             "name": "nabirds",
             "base_path": "/home/brust/datasets/nabirds",
             "base_path": "/home/brust/datasets/nabirds",
@@ -58,11 +63,11 @@ def main():
     # Add relation source
     # Add relation source
     experiment_container.knowledge_base.add_hyponymy_relation([dataset.get_hyponymy_relation_source()])
     experiment_container.knowledge_base.add_hyponymy_relation([dataset.get_hyponymy_relation_source()])
 
 
-    exit
-
     base_model = experiment_container.base_model
     base_model = experiment_container.base_model
     training_samples = dataset.train_pool(0, "label_gt")
     training_samples = dataset.train_pool(0, "label_gt")
-    base_model.observe(training_samples, "label_gt")
+    training_samples = experiment_container.interactor.query_annotations_for(training_samples, "label_gt", "label_ann")
+
+    base_model.observe(training_samples, "label_ann")
 
 
     test_samples = dataset.test_pool(0, "label_gt")[:100]
     test_samples = dataset.test_pool(0, "label_gt")[:100]
     test_samples = base_model.predict(test_samples, "label_pred")
     test_samples = base_model.predict(test_samples, "label_pred")