How to use HydroEmbed/HydroEmbed-FITB-MiniLM-DualLoss with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("HydroEmbed/HydroEmbed-FITB-MiniLM-DualLoss") sentences = [ "The groundwater residence time in the Chashma-Mianwali area was estimated to be in the range of _____ years based on Tracer Lump Parameter Model (LPM) and Chlorofluorocarbons (CFCs) data.", "14–59", "point scale; radar data", "dissolved oxygen" ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [4, 4]
