public code v1

2026-05-22 10:02:10 +02:00
commit 46a9ecf065
166 changed files with 6982454 additions and 0 deletions
@@ -0,0 +1,96 @@
+base:
+  data:
+    groups:
+      filepath_or_buffer:
+          - 'datasets/stratigis/groupsWithHighRatings5.txt'
+          - 'datasets/stratigis/groupsWithModerateRatings5untested.txt'
+    testdata:
+      filepath_or_buffer: 'datasets/fake_data.csv'
+      sep: '\t'
+      skiprows: 1
+      names: [ 'userId', 'itemId', 'rating', 'timestamp']
+    test:
+      filepath_or_buffer: 'datasets/stratigis/ratings.csv'
+      sep: ','
+      skiprows: 1
+      names: [ 'userId', 'itemId', 'rating', 'timestamp']
+    ml32m:
+      filepath_or_buffer: 'datasets/ml-32m/ratings.csv'
+      sep: ','
+      skiprows: 1
+      names: [ 'userId', 'itemId', 'rating', 'timestamp']
+    ml100k:
+        filepath_or_buffer: 'datasets/ml-100k/u.data'
+        sep: '\t'
+        skiprows: 0
+        names: [ 'userId', 'itemId', 'rating', 'timestamp']
+    ml1m:
+      filepath_or_buffer: 'datasets/ml-1m/ratings.dat'
+      sep: '::'
+      names: [ 'userId', 'itemId', 'rating', 'timestamp' ]
+    tags:
+        tags_file: 'datasets/stratigis/tags.csv'
+  model:
+    gmf:
+      learning_rate: 0.005
+      weight_decay: 0.0000001
+      latent_dim: 8
+      epochs: 30
+      num_negative: 10
+      batch_size: 1024
+      cuda: False
+      optimizer_name: 'adam'
+    mlp:
+      learning_rate: 0.005
+      weight_decay: 0.0000001
+      latent_dim: 8
+      epochs: 30
+      num_negative: 10
+      batch_size: 1024
+      cuda: False
+      optimizer_name: 'adam'
+    als:
+      learning_rate: 0.1
+      latent_dim: 100
+      epochs: 10
+      reg_term: 0.001
+
+    bpr:
+      learning_rate: 0.01
+      latent_dim: 100
+      epochs: 10
+      reg_term: 0.001
+    emf:
+      learning_rate: 0.01
+      reg_term: 0.001
+      expl_reg_term: 0.0
+      latent_dim: 80
+      epochs: 10
+      positive_threshold: 3
+      knn: 10
+    mf:
+      learning_rate: 0.01
+      reg_term: 0.001
+      expl_reg_term: 0.0
+      latent_dim: 80
+      epochs: 10
+      positive_threshold: 3
+      knn: 10
+    autoencoder:
+      learning_rate: 0.005
+      weight_decay: 0.0000001
+      hidden_layer_features: 8
+      epochs: 30
+      cuda: False
+      optimizer_name: 'adam'
+      positive_threshold: 3
+      knn: 10
+      expl: true
+  explainer:
+    lore4groups:
+      n_similar_for_tree: 100
+      rating_threshold_for_like: 3.0
+      max_tree_depth: 5
+      top_n_labels: 5000
+      min_rating_for_history: 1.0
+      similarity_threshold: 0.1