from sentence_transformers import SentenceTransformer

model = SentenceTransformer("Talina06/arxiv-search")

query = "Neural networks in medicine"
query_embedding = model.encode(query)

# Use FAISS or cosine similarity to retrieve similar papers

Training Details

Training Data: 100k+ Arxiv research papers
Training Framework: Sentence Transformers
Hyperparameters:
- Learning Rate: 2e-5
- Batch Size: 100
- Epochs: 10
Hardware Used: TPU & GPU

Example Search Results

Query	Top Matching Paper Title	Similarity Score
"Neural networks in healthcare"	"Deep Learning for Medical Diagnosis"	0.89
"Quantum cryptography"	"A Survey on Quantum-Safe Encryption"	0.87