add model

analogy.json

@@ -0,0 +1 @@
+{"distance_function": "cosine_similarity", "sat/test": 0.4599406528189911, "sat/valid": 0.43243243243243246, "u2/test": 0.4166666666666667, "u2/valid": 0.3333333333333333, "u4/test": 0.4375, "u4/valid": 0.3958333333333333, "google/test": 0.84, "google/valid": 0.86, "bats/test": 0.7326292384658143, "bats/valid": 0.7638190954773869, "sat_full": 0.4572192513368984}

classification.json

@@ -0,0 +1 @@
+{"lexical_relation_classification/BLESS": {"classifier_config": {"activation": "relu", "alpha": 0.0001, "batch_size": "auto", "beta_1": 0.9, "beta_2": 0.999, "early_stopping": false, "epsilon": 1e-08, "hidden_layer_sizes": [100], "learning_rate": "constant", "learning_rate_init": 0.001, "max_fun": 15000, "max_iter": 200, "momentum": 0.9, "n_iter_no_change": 10, "nesterovs_momentum": true, "power_t": 0.5, "random_state": 0, "shuffle": true, "solver": "adam", "tol": 0.0001, "validation_fraction": 0.1, "verbose": false, "warm_start": false}, "test/accuracy": 0.9058309477173422, "test/f1_macro": 0.89974713256054, "test/f1_micro": 0.9058309477173422, "test/p_macro": 0.887765109071328, "test/p_micro": 0.9058309477173422, "test/r_macro": 0.9129271192123373, "test/r_micro": 0.9058309477173422}, "lexical_relation_classification/CogALexV": {"classifier_config": {"activation": "relu", "alpha": 0.0001, "batch_size": "auto", "beta_1": 0.9, "beta_2": 0.999, "early_stopping": false, "epsilon": 1e-08, "hidden_layer_sizes": [100], "learning_rate": "constant", "learning_rate_init": 0.001, "max_fun": 15000, "max_iter": 200, "momentum": 0.9, "n_iter_no_change": 10, "nesterovs_momentum": true, "power_t": 0.5, "random_state": 0, "shuffle": true, "solver": "adam", "tol": 0.0001, "validation_fraction": 0.1, "verbose": false, "warm_start": false}, "test/accuracy": 0.826056338028169, "test/f1_macro": 0.6201374746332642, "test/f1_micro": 0.826056338028169, "test/p_macro": 0.6517885336110181, "test/p_micro": 0.826056338028169, "test/r_macro": 0.5964780161213106, "test/r_micro": 0.826056338028169}, "lexical_relation_classification/EVALution": {"classifier_config": {"activation": "relu", "alpha": 0.0001, "batch_size": "auto", "beta_1": 0.9, "beta_2": 0.999, "early_stopping": false, "epsilon": 1e-08, "hidden_layer_sizes": [100], "learning_rate": "constant", "learning_rate_init": 0.001, "max_fun": 15000, "max_iter": 200, "momentum": 0.9, "n_iter_no_change": 10, "nesterovs_momentum": true, "power_t": 0.5, "random_state": 0, "shuffle": true, "solver": "adam", "tol": 0.0001, "validation_fraction": 0.1, "verbose": false, "warm_start": false}, "test/accuracy": 0.6457204767063922, "test/f1_macro": 0.6243022596465048, "test/f1_micro": 0.6457204767063922, "test/p_macro": 0.6422972967692119, "test/p_micro": 0.6457204767063922, "test/r_macro": 0.6123731242708812, "test/r_micro": 0.6457204767063922}, "lexical_relation_classification/K&H+N": {"classifier_config": {"activation": "relu", "alpha": 0.0001, "batch_size": "auto", "beta_1": 0.9, "beta_2": 0.999, "early_stopping": false, "epsilon": 1e-08, "hidden_layer_sizes": [100], "learning_rate": "constant", "learning_rate_init": 0.001, "max_fun": 15000, "max_iter": 200, "momentum": 0.9, "n_iter_no_change": 10, "nesterovs_momentum": true, "power_t": 0.5, "random_state": 0, "shuffle": true, "solver": "adam", "tol": 0.0001, "validation_fraction": 0.1, "verbose": false, "warm_start": false}, "test/accuracy": 0.9559017875773805, "test/f1_macro": 0.8816731086105152, "test/f1_micro": 0.9559017875773805, "test/p_macro": 0.9227793107598984, "test/p_micro": 0.9559017875773805, "test/r_macro": 0.8575980124222238, "test/r_micro": 0.9559017875773805}, "lexical_relation_classification/ROOT09": {"classifier_config": {"activation": "relu", "alpha": 0.0001, "batch_size": "auto", "beta_1": 0.9, "beta_2": 0.999, "early_stopping": false, "epsilon": 1e-08, "hidden_layer_sizes": [100], "learning_rate": "constant", "learning_rate_init": 0.001, "max_fun": 15000, "max_iter": 200, "momentum": 0.9, "n_iter_no_change": 10, "nesterovs_momentum": true, "power_t": 0.5, "random_state": 0, "shuffle": true, "solver": "adam", "tol": 0.0001, "validation_fraction": 0.1, "verbose": false, "warm_start": false}, "test/accuracy": 0.8862425571921028, "test/f1_macro": 0.8841357906198278, "test/f1_micro": 0.8862425571921027, "test/p_macro": 0.8849639198744838, "test/p_micro": 0.8862425571921028, "test/r_macro": 0.8833813015044947, "test/r_micro": 0.8862425571921028}}

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "roberta-base",
+  "_name_or_path": "relbert-roberta-base-semeval2012-v6-average-prompt-b-nce-0",
   "architectures": [
     "RobertaModel"
   ],

relation_mapping.json

@@ -0,0 +1 @@
+{"accuracy": 0.7995436507936508, "prediction": [{"source": ["solar system", "sun", "planet", "mass", "attracts", "revolves", "gravity"], "true": ["atom", "nucleus", "electron", "charge", "attracts", "revolves", "electromagnetism"], "pred": ["electromagnetism", "electron", "nucleus", "charge", "attracts", "revolves", "atom"], "alignment_match": false, "accuracy": 0.42857142857142855, "similarity": 0.9576257983414723, "similarity_true": 0.9504084897997144}, {"source": ["water", "flows", "pressure", "water tower", "bucket", "filling", "emptying", "hydrodynamics"], "true": ["heat", "transfers", "temperature", "burner", "kettle", "heating", "cooling", "thermodynamics"], "pred": ["heat", "transfers", "temperature", "burner", "kettle", "heating", "cooling", "thermodynamics"], "alignment_match": true, "accuracy": 1, "similarity": 0.8143134047711483, "similarity_true": 0.8143134047711483}, {"source": ["waves", "shore", "reflects", "water", "breakwater", "rough", "calm", "crashing"], "true": ["sounds", "wall", "echoes", "air", "insulation", "loud", "quiet", "vibrating"], "pred": ["sounds", "insulation", "echoes", "air", "wall", "loud", "quiet", "vibrating"], "alignment_match": false, "accuracy": 0.75, "similarity": 0.8118570767721925, "similarity_true": 0.8078544898111648}, {"source": ["combustion", "fire", "fuel", "burning", "hot", "intense", "oxygen", "carbon dioxide"], "true": ["respiration", "animal", "food", "breathing", "living", "vigorous", "oxygen", "carbon dioxide"], "pred": ["respiration", "animal", "food", "breathing", "living", "vigorous", "oxygen", "carbon dioxide"], "alignment_match": true, "accuracy": 1, "similarity": 0.9240900159119896, "similarity_true": 0.9240900159119896}, {"source": ["sound", "low", "high", "echoes", "loud", "quiet", "horn"], "true": ["light", "red", "violet", "reflects", "bright", "dim", "lens"], "pred": ["violet", "dim", "red", "reflects", "bright", "light", "lens"], "alignment_match": false, "accuracy": 0.42857142857142855, "similarity": 0.8169790414502984, "similarity_true": 0.8064887534056607}, {"source": ["projectile", "trajectory", "earth", "parabolic", "air", "gravity", "attracts"], "true": ["planet", "orbit", "sun", "elliptical", "space", "gravity", "attracts"], "pred": ["sun", "orbit", "planet", "elliptical", "space", "gravity", "attracts"], "alignment_match": false, "accuracy": 0.7142857142857143, "similarity": 0.9648638939921251, "similarity_true": 0.9579000495798417}, {"source": ["breeds", "selection", "conformance", "artificial", "popularity", "breeding", "domesticated"], "true": ["species", "competition", "adaptation", "natural", "fitness", "mating", "wild"], "pred": ["species", "competition", "adaptation", "wild", "fitness", "mating", "natural"], "alignment_match": false, "accuracy": 0.7142857142857143, "similarity": 0.8350827729877157, "similarity_true": 0.8337754045448718}, {"source": ["ball", "billiards", "speed", "table", "bouncing", "moving", "slow", "fast"], "true": ["molecules", "gas", "temperature", "container", "pressing", "moving", "cold", "hot"], "pred": ["molecules", "gas", "temperature", "container", "cold", "moving", "hot", "pressing"], "alignment_match": false, "accuracy": 0.625, "similarity": 0.8963198028072434, "similarity_true": 0.8940494110357722}, {"source": ["computer", "processing", "erasing", "write", "read", "memory", "outputs", "inputs", "bug"], "true": ["mind", "thinking", "forgetting", "memorize", "remember", "memory", "muscles", "senses", "mistake"], "pred": ["mind", "muscles", "forgetting", "remember", "memorize", "memory", "senses", "thinking", "mistake"], "alignment_match": false, "accuracy": 0.4444444444444444, "similarity": 0.8389583951825731, "similarity_true": 0.8277314351393401}, {"source": ["slot machines", "reels", "spinning", "winning", "losing"], "true": ["bacteria", "genes", "mutating", "reproducing", "dying"], "pred": ["bacteria", "genes", "mutating", "reproducing", "dying"], "alignment_match": true, "accuracy": 1, "similarity": 0.7846841262187907, "similarity_true": 0.7846841262187907}, {"source": ["war", "soldier", "destroy", "fighting", "defeat", "attacks", "weapon"], "true": ["argument", "debater", "refute", "arguing", "acceptance", "criticizes", "logic"], "pred": ["argument", "debater", "refute", "arguing", "logic", "criticizes", "acceptance"], "alignment_match": false, "accuracy": 0.7142857142857143, "similarity": 0.80424872183556, "similarity_true": 0.8040981034675441}, {"source": ["buyer", "merchandise", "buying", "selling", "returning", "valuable", "worthless"], "true": ["believer", "belief", "accepting", "advocating", "rejecting", "true", "false"], "pred": ["believer", "rejecting", "belief", "advocating", "accepting", "true", "false"], "alignment_match": false, "accuracy": 0.5714285714285714, "similarity": 0.7954272104101929, "similarity_true": 0.7905947652686135}, {"source": ["foundations", "buildings", "supporting", "solid", "weak", "crack"], "true": ["reasons", "theories", "confirming", "rational", "dubious", "flaw"], "pred": ["reasons", "theories", "confirming", "rational", "dubious", "flaw"], "alignment_match": true, "accuracy": 1, "similarity": 0.7812446948067134, "similarity_true": 0.7812446948067134}, {"source": ["obstructions", "destination", "route", "traveller", "traveling", "companion", "arriving"], "true": ["difficulties", "goal", "plan", "person", "problem solving", "partner", "succeeding"], "pred": ["difficulties", "goal", "plan", "person", "problem solving", "partner", "succeeding"], "alignment_match": true, "accuracy": 1, "similarity": 0.8404202837864979, "similarity_true": 0.8404202837864979}, {"source": ["money", "allocate", "budget", "effective", "cheap", "expansive"], "true": ["time", "invest", "schedule", "efficient", "quick", "slow"], "pred": ["time", "invest", "schedule", "efficient", "quick", "slow"], "alignment_match": true, "accuracy": 1, "similarity": 0.8213253233120695, "similarity_true": 0.8213253233120695}, {"source": ["seeds", "planted", "fruitful", "fruit", "grow", "wither", "blossom"], "true": ["ideas", "inspired", "productive", "product", "develop", "fail", "succeed"], "pred": ["ideas", "inspired", "productive", "product", "develop", "fail", "succeed"], "alignment_match": true, "accuracy": 1, "similarity": 0.8534395271969207, "similarity_true": 0.8534395271969207}, {"source": ["machine", "working", "turned on", "turned off", "broken", "power", "repair"], "true": ["mind", "thinking", "awake", "asleep", "confused", "intelligence", "therapy"], "pred": ["mind", "thinking", "awake", "asleep", "confused", "intelligence", "therapy"], "alignment_match": true, "accuracy": 1, "similarity": 0.7717230414332904, "similarity_true": 0.7717230414332904}, {"source": ["object", "hold", "weight", "heavy", "light"], "true": ["idea", "understand", "analyze", "important", "trivial"], "pred": ["idea", "understand", "analyze", "trivial", "important"], "alignment_match": false, "accuracy": 0.6, "similarity": 0.7454507962400786, "similarity_true": 0.7417522947794583}, {"source": ["follow", "leader", "path", "follower", "lost", "wanders", "twisted", "straight"], "true": ["understand", "speaker", "argument", "listener", "misunderstood", "digresses", "complicated", "simple"], "pred": ["understand", "speaker", "argument", "listener", "misunderstood", "digresses", "complicated", "simple"], "alignment_match": true, "accuracy": 1, "similarity": 0.8153576261327493, "similarity_true": 0.8153576261327493}, {"source": ["seeing", "light", "illuminating", "darkness", "view", "hidden"], "true": ["understanding", "knowledge", "explaining", "confusion", "interpretation", "secret"], "pred": ["understanding", "knowledge", "explaining", "confusion", "interpretation", "secret"], "alignment_match": true, "accuracy": 1, "similarity": 0.7698682574790119, "similarity_true": 0.7698682574790119}]}