anno-eval 0.8.0

Evaluation harnesses, datasets, and muxer-backed sampling for anno
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

//! Scientific/academic domain synthetic data.

use super::super::types::helpers::entity;
use super::super::types::{AnnotatedExample, Difficulty, Domain};
use anno_core::EntityType;

/// Scientific/academic domain dataset.
pub fn dataset() -> Vec<AnnotatedExample> {
    vec![
        AnnotatedExample {
            text: "Dr. Jennifer Doudna won the Nobel Prize for CRISPR research at UC Berkeley.".into(),
            entities: vec![
                entity("Dr. Jennifer Doudna", EntityType::Person, 0),
                entity("UC Berkeley", EntityType::Organization, 63),
            ],
            domain: Domain::Scientific,
            difficulty: Difficulty::Medium,
        },
        AnnotatedExample {
            text: "NASA's James Webb Space Telescope captured images of the Carina Nebula.".into(),
            entities: vec![
                entity("NASA", EntityType::Organization, 0),
                entity("Carina Nebula", EntityType::Location, 57),
            ],
            domain: Domain::Scientific,
            difficulty: Difficulty::Hard,
        },
        AnnotatedExample {
            text: "NASA launched Artemis from Cape Canaveral, Florida on November 16, 2022.".into(),
            entities: vec![
                entity("NASA", EntityType::Organization, 0),
                entity("Cape Canaveral", EntityType::Location, 27),
                entity("Florida", EntityType::Location, 43),
                entity("November 16, 2022", EntityType::Date, 54),
            ],
            domain: Domain::Scientific,
            difficulty: Difficulty::Medium,
        },
        AnnotatedExample {
            text:
                "Einstein published special relativity while working at the Swiss Patent Office in Bern."
                    .into(),
            entities: vec![
                entity("Einstein", EntityType::Person, 0),
                entity("Swiss Patent Office", EntityType::Organization, 59),
                entity("Bern", EntityType::Location, 82),
            ],
            domain: Domain::Scientific,
            difficulty: Difficulty::Medium,
        },
        AnnotatedExample {
            text: "CERN's Large Hadron Collider near Geneva discovered the Higgs boson.".into(),
            entities: vec![
                entity("CERN", EntityType::Organization, 0),
                entity("Geneva", EntityType::Location, 34),
            ],
            domain: Domain::Scientific,
            difficulty: Difficulty::Medium,
        },
        AnnotatedExample {
            text: "Marie Curie conducted radioactivity research at the University of Paris.".into(),
            entities: vec![
                entity("Marie Curie", EntityType::Person, 0),
                entity("University of Paris", EntityType::Organization, 52),
            ],
            domain: Domain::Scientific,
            difficulty: Difficulty::Easy,
        },
        AnnotatedExample {
            text: "The Mars Perseverance rover landed in Jezero Crater on February 18, 2021.".into(),
            entities: vec![
                entity("Jezero Crater", EntityType::Location, 38),
                entity("February 18, 2021", EntityType::Date, 55),
            ],
            domain: Domain::Scientific,
            difficulty: Difficulty::Medium,
        },
        AnnotatedExample {
            text: "DeepMind's AlphaFold predicted 200 million protein structures.".into(),
            entities: vec![entity("DeepMind", EntityType::Organization, 0)],
            domain: Domain::Scientific,
            difficulty: Difficulty::Easy,
        },
        AnnotatedExample {
            text: "Prof. Katalin Karikó received the Nobel Prize for mRNA vaccine research.".into(),
            entities: vec![entity("Prof. Katalin Karikó", EntityType::Person, 0)],
            domain: Domain::Scientific,
            difficulty: Difficulty::Easy,
        },
        AnnotatedExample {
            text: "The Voyager 1 spacecraft, launched in 1977, is now 15 billion miles from Earth."
                .into(),
            entities: vec![entity("Earth", EntityType::Location, 73)],
            domain: Domain::Scientific,
            difficulty: Difficulty::Hard,
        },
    ]
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_scientific_dataset_not_empty() {
        assert!(!dataset().is_empty());
    }
}