Downloads · 30 days
0
qpn2vec_step1
hiki-t/qpn2vec_step1
qpn2vec_step1 is a machine learning model from hiki-t. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
model-step-1 query+doc(pos/neg) - vec - cossim
Access
Public
Updated Jun 17, 2025
Repo size
14.9 KB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth14.9 KB · 88%
At a glance
- Access
- Public
- Created
- Jun 17, 2025
- Updated
- Jun 17, 2025
- SHA
- 563fc07d
Model card
From the Hugging Face model README
Custom sentence2vec projection model
so far
model-step-1
query+doc(pos/neg) -> vec -> cos_sim
train a model for loss of cos_sim btw q-n, q-p with margin
device: cpu
epoch size: 3
input_dim or emb dim: 25
projection_dim: 128
margin=0.2
batch size: 32
neg doc size: 10 rows
overall training size: 82_325
time spent to run: 29m49s
Epoch 1, Loss: 0.0960
Epoch 2, Loss: 0.0559
Epoch 3, Loss: 0.0526
- Created
- Jun 17, 2025
- Updated
- Jun 17, 2025