sentence-transformers How to use dkcodes/poly-headline with sentence-transformers:
from sentence_transformers import SentenceTransformer
model = SentenceTransformer("dkcodes/poly-headline")
sentences = [
"USGS reports all earthquakes below magnitude 8.0 this quarter",
"Megaquake by September 30? A \"megaquake\" is defined as an earthquake with a magnitude of 8.0 or greater. This market will resolve to “Yes” if 1 or more earthquakes with a magnitude of 8.0 or higher occur anywhere on Earth between July 30 and September 30, 2025, 11:59 PM ET. Otherwise, this market will resolve to “No”. The resolution source for this market is the United States Geological Survey (USGS) Earthquake Hazards Program (https://earthquake.usgs.gov/earthquakes/browse/significant.php#sigdef). If an earthquake of substantial size has occurred within this market's timeframe but not yet appeared on the resolution source, this market may remain open until October 7, 2025, 11:59 PM ET, or until the earthquake in question otherwise appears on the resolution source. If such an earthquake has not appeared on the resolution source by that date, another credible resolution source will be used. After a qualifying earthquake is registered, this market will remain open for 24 hours to account for any revisions to its recorded magnitude. After 24 hours, this market will resolve according to the latest provided data.",
"Will \"Elio\" Opening Weekend Box Office be less than $20m? This market will resolve according to how much “Elio” (2025) will gross domestically on its opening weekend. The “Box Office” https://www.the-numbers.com/movie/Elio(2025)#tab=box-office will be used to resolve this market once the values for the 3-day opening weekend (June 20 - June 22) are final (i.e. not studio estimates). If the reported value falls exactly between two brackets, then this market will resolve to the higher range bracket. Please note, this market will resolve according to the The Numbers figures provided under Weekend Box Office Performance for the 3-day weekend (which typically includes Thursday's previews), regardless of whether domestic refers to only the USA, or to USA and Canada, etc. If there is no final data available by June 30, 2025, 11:59 PM ET, another credible resolution source will be chosen.",
"Researchers Explore Correlation Between Solar Activity and Seismic Events"
]
embeddings = model.encode(sentences)
similarities = model.similarity(embeddings, embeddings)
print(similarities.shape)
# [4, 4] 
