1from sentence_transformers import CrossEncoder
2
3# Download from the 🤗 Hub
4model = CrossEncoder("cross_encoder_model_id")
5# Get scores for pairs of texts
6pairs = [
7 ["In which year did the publisher of Roald Dahl's Guide to Railway Safety cease to exist?", "Glacier Park International Airport is in Flathead County, Montana, six miles northeast of Kalispell. The airport is owned and operated by the Flathead Municipal Airport Authority, a public agency created by the county in 1974.</s>Roald Dahl's Guide to Railway Safety was published in 1991 by the British Railways Board. The British Railways Board had asked Roald Dahl to write the text of the booklet, and Quentin Blake to illustrate it, to help young people enjoy using the railways safely.</s>The British Railways Board (BRB) was a nationalised industry in the United Kingdom that operated from 1963 to 2001. Until 1997 it was responsible for most railway services in Great Britain, trading under the brand name British Railways and, from 1965, British Rail. It did not operate railways in Northern Ireland, where railways were the responsibility of the Government of Northern Ireland."],
8 ['Who is the brother of the developer of Buffy the Animated Series?', "Waylon Malloy Payne (born April 5, 1972) is an American country singer, songwriter, musician and actor. He is the son of the country singer Sammi Smith.</s>WVFM, known simply as FM 106.5 and formerly WQLR, is a Classic Hits-leaning Adult Contemporary outlet serving the Kalamazoo, Michigan radio market. The station's frequency is 106.5\xa0MHz on the FM dial with an ERP of 33\xa0kW. They are owned by Midwest Communications. WVFM 106.5 is located on a crowded Frequency across south-west lower Michigan. The station covers all of Kalamazoo/Battle Creek area, can be heard well in the Grand Rapids area, and reaches as far north and east as Lansing and Jackson. During summer months, the station can be received to Flint and Ann Arbor on occasion.</s>Fred Iltis (Brno, Czechoslovakia, April 20, 1923 – San Jose, California, December 11, 2008) was an American entomologist. His research focused on the biosystematics and life cycle of mosquitoes."],
9 ['Who did the performer of So Under Pressure play in Home and Away?', 'The Acoma Massacre was fought in January 1599 between Spanish conquistadors and Acoma Native Americans in what is now New Mexico. After twelve soldiers were killed at Acoma Pueblo in 1598, the Spanish retaliated by launching a punitive expedition, which led to the deaths of around 800 men, women and children during a three - day battle. Several hundred survivors were also enslaved or otherwise severely punished.</s>The NBA Championship ring is an annual award given by the National Basketball Association to the team that wins the NBA Finals. Rings are presented to the team\'s players, coaches, and members of the executive front office. Red Auerbach has the most rings overall with 16. Phil Jackson has the most as coach and Bill Russell has the most as a player (11 each)</s>Tibetan sources say Deshin Shekpa also persuaded the Yongle Emperor not to impose his military might on Tibet as the Mongols had previously done. Thinley writes that before the Karmapa returned to Tibet, the Yongle Emperor began planning to send a military force into Tibet to forcibly give the Karmapa authority over all the Tibetan Buddhist schools but Deshin Shekpa dissuaded him. However, Hok-Lam Chan states that "there is little evidence that this was ever the emperor\'s intention" and that evidence indicates that Deshin Skekpa was invited strictly for religious purposes.'],
10 ["Big Bill Morganfield's father is associated with which subgenre of the blues?", 'William "Big Bill" Morganfield (born June 19, 1956) is an American blues singer and guitarist, who is the son of Muddy Waters.</s>WNZK is a radio station in Dearborn Heights, Michigan, United States. It began broadcasting October 12, 1985. It broadcasts in the AM radio band at 690 kHz during the daytime and at 680\xa0kHz at night. This is to protect the nighttime pattern of Montreal, Quebec\'s CKGM, a clear-channel station on 690. WNZK is the only North American AM station to broadcast on two frequencies, at least according to the FCC online database.</s>McKinley Morganfield (April 4, 1913 -- April 30, 1983), known professionally as Muddy Waters, was an American blues musician who is often cited as the ``father of modern Chicago blues \'\'.'],
11 ['Who was played by the Nothing That You Are performer in Princess Diaries?', 'The Madonna of Chancellor Rolin is an oil painting by the Early Netherlandish master Jan van Eyck, dating from around 1435. It is kept in the Musée du Louvre, Paris, and was commissioned by Nicolas Rolin, aged 60, chancellor of the Duchy of Burgundy, whose votive portrait takes up the left side of the picture, for his parish church, "Notre-Dame-du-Chastel" in Autun, where it remained until the church burnt down in 1793. After a period in Autun Cathedral, it was moved to the Louvre in 1805.</s>Jon Monday (born 1947 in San Jose, California) is an American producer and distributor of CDs and DVDs across an eclectic range of material such as Swami Prabhavananda, Aldous Huxley, Christopher Isherwood, Huston Smith, Chalmers Johnson, and Charles Bukowski. Monday directed and co-produced with Jennifer Douglas the feature-length documentary "Save KLSD: Media Consolidation and Local Radio". He is also President of Benchmark Recordings, which owns and distributes the early catalog of The Fabulous Thunderbirds CDs and a live recording of Mike Bloomfield.</s>Strictly Inc. is the self-titled project album, released by Genesis keyboardist Tony Banks, and Wang Chung lead vocalist Jack Hues, in 1995 on Virgin Records. Tony Banks wanted the album release—as the title suggested—with no reference to the band members; but the record company went against his wishes. This was Banks\' fifth studio album (his second issued under a band name and seventh album overall).'],
12]
13scores = model.predict(pairs)
14print(scores.shape)
15# (5,)
16
17# Or rank different texts based on similarity to a single text
18ranks = model.rank(
19 "In which year did the publisher of Roald Dahl's Guide to Railway Safety cease to exist?",
20 [
21 "Glacier Park International Airport is in Flathead County, Montana, six miles northeast of Kalispell. The airport is owned and operated by the Flathead Municipal Airport Authority, a public agency created by the county in 1974.</s>Roald Dahl's Guide to Railway Safety was published in 1991 by the British Railways Board. The British Railways Board had asked Roald Dahl to write the text of the booklet, and Quentin Blake to illustrate it, to help young people enjoy using the railways safely.</s>The British Railways Board (BRB) was a nationalised industry in the United Kingdom that operated from 1963 to 2001. Until 1997 it was responsible for most railway services in Great Britain, trading under the brand name British Railways and, from 1965, British Rail. It did not operate railways in Northern Ireland, where railways were the responsibility of the Government of Northern Ireland.",
22 "Waylon Malloy Payne (born April 5, 1972) is an American country singer, songwriter, musician and actor. He is the son of the country singer Sammi Smith.</s>WVFM, known simply as FM 106.5 and formerly WQLR, is a Classic Hits-leaning Adult Contemporary outlet serving the Kalamazoo, Michigan radio market. The station's frequency is 106.5\xa0MHz on the FM dial with an ERP of 33\xa0kW. They are owned by Midwest Communications. WVFM 106.5 is located on a crowded Frequency across south-west lower Michigan. The station covers all of Kalamazoo/Battle Creek area, can be heard well in the Grand Rapids area, and reaches as far north and east as Lansing and Jackson. During summer months, the station can be received to Flint and Ann Arbor on occasion.</s>Fred Iltis (Brno, Czechoslovakia, April 20, 1923 – San Jose, California, December 11, 2008) was an American entomologist. His research focused on the biosystematics and life cycle of mosquitoes.",
23 'The Acoma Massacre was fought in January 1599 between Spanish conquistadors and Acoma Native Americans in what is now New Mexico. After twelve soldiers were killed at Acoma Pueblo in 1598, the Spanish retaliated by launching a punitive expedition, which led to the deaths of around 800 men, women and children during a three - day battle. Several hundred survivors were also enslaved or otherwise severely punished.</s>The NBA Championship ring is an annual award given by the National Basketball Association to the team that wins the NBA Finals. Rings are presented to the team\'s players, coaches, and members of the executive front office. Red Auerbach has the most rings overall with 16. Phil Jackson has the most as coach and Bill Russell has the most as a player (11 each)</s>Tibetan sources say Deshin Shekpa also persuaded the Yongle Emperor not to impose his military might on Tibet as the Mongols had previously done. Thinley writes that before the Karmapa returned to Tibet, the Yongle Emperor began planning to send a military force into Tibet to forcibly give the Karmapa authority over all the Tibetan Buddhist schools but Deshin Shekpa dissuaded him. However, Hok-Lam Chan states that "there is little evidence that this was ever the emperor\'s intention" and that evidence indicates that Deshin Skekpa was invited strictly for religious purposes.',
24 'William "Big Bill" Morganfield (born June 19, 1956) is an American blues singer and guitarist, who is the son of Muddy Waters.</s>WNZK is a radio station in Dearborn Heights, Michigan, United States. It began broadcasting October 12, 1985. It broadcasts in the AM radio band at 690 kHz during the daytime and at 680\xa0kHz at night. This is to protect the nighttime pattern of Montreal, Quebec\'s CKGM, a clear-channel station on 690. WNZK is the only North American AM station to broadcast on two frequencies, at least according to the FCC online database.</s>McKinley Morganfield (April 4, 1913 -- April 30, 1983), known professionally as Muddy Waters, was an American blues musician who is often cited as the ``father of modern Chicago blues \'\'.',
25 'The Madonna of Chancellor Rolin is an oil painting by the Early Netherlandish master Jan van Eyck, dating from around 1435. It is kept in the Musée du Louvre, Paris, and was commissioned by Nicolas Rolin, aged 60, chancellor of the Duchy of Burgundy, whose votive portrait takes up the left side of the picture, for his parish church, "Notre-Dame-du-Chastel" in Autun, where it remained until the church burnt down in 1793. After a period in Autun Cathedral, it was moved to the Louvre in 1805.</s>Jon Monday (born 1947 in San Jose, California) is an American producer and distributor of CDs and DVDs across an eclectic range of material such as Swami Prabhavananda, Aldous Huxley, Christopher Isherwood, Huston Smith, Chalmers Johnson, and Charles Bukowski. Monday directed and co-produced with Jennifer Douglas the feature-length documentary "Save KLSD: Media Consolidation and Local Radio". He is also President of Benchmark Recordings, which owns and distributes the early catalog of The Fabulous Thunderbirds CDs and a live recording of Mike Bloomfield.</s>Strictly Inc. is the self-titled project album, released by Genesis keyboardist Tony Banks, and Wang Chung lead vocalist Jack Hues, in 1995 on Virgin Records. Tony Banks wanted the album release—as the title suggested—with no reference to the band members; but the record company went against his wishes. This was Banks\' fifth studio album (his second issued under a band name and seventh album overall).',
26 ]
27)
28# [{'corpus_id': ..., 'score': ...}, {'corpus_id': ..., 'score': ...}, ...]
