Spaces:

PECCAVI-TEXT
/

peccavi

Sleeping

PECCAVI-TEXT commited on May 19

Commit

d57a0cf

verified ·

1 Parent(s): f1d5ed0

Update renderers/tree.py

Files changed (1) hide show

renderers/tree.py CHANGED Viewed

@@ -144,9 +144,9 @@ def generate_subplot1(paraphrased_sentence, masked_sentences, strategies, highli
     # Define the text for each edge
     default_edge_texts = [
         "Highest Entropy Masking", "Pseudo-random Masking", "Random Masking",
-        "Greedy Sampling", "Temperature Sampling", "Exponential Minimum Sampling",
-        "Inverse Transform Sampling", "Greedy Sampling", "Temperature Sampling",
-        "Exponential Minimum Sampling", "Inverse Transform Sampling", "Greedy Sampling",
         "Temperature Sampling", "Exponential Minimum Sampling", "Inverse Transform Sampling"
     ]
@@ -248,10 +248,11 @@ def generate_subplot2(masked_sentences, sampled_sentences, highlight_info, commo
     """
     # Define sampling techniques
     sampling_techniques = [
-        "Greedy Sampling",
-        "Temperature Sampling",
-        "Exponential Minimum Sampling",
-        "Inverse Transform Sampling"
     ]
     # Calculate total number of nodes

     # Define the text for each edge
     default_edge_texts = [
         "Highest Entropy Masking", "Pseudo-random Masking", "Random Masking",
+        "Greedy Sampling", "Tournament Sampling", "Temperature Sampling", "Exponential Minimum Sampling",
+        "Inverse Transform Sampling", "Greedy Sampling", "Tournament Sampling", "Temperature Sampling",
+        "Exponential Minimum Sampling", "Inverse Transform Sampling", "Greedy Sampling", "Tournament Sampling",
         "Temperature Sampling", "Exponential Minimum Sampling", "Inverse Transform Sampling"
     ]
     """
     # Define sampling techniques
     sampling_techniques = [
+        "Inverse Transform Sampling",
+        "Exponential Minimum Sampling",
+        "Temperature Sampling",
+        "Greedy Sampling",
+        "Tournament Sampling",
     ]
     # Calculate total number of nodes