backend/passfinder/recomendations/service/models/models.py

27 lines
817 B
Python
Raw Normal View History

2023-05-27 11:13:41 +03:00
from annoy import AnnoyIndex
N_DIMENSIONAL = 768
2023-05-28 00:11:44 +03:00
attracion_model = AnnoyIndex(N_DIMENSIONAL, "angular")
attracion_model.load("passfinder/recomendations/service/models/dost.ann")
2023-05-27 11:13:41 +03:00
2023-05-28 00:11:44 +03:00
cinema_model = AnnoyIndex(N_DIMENSIONAL, "angular")
cinema_model.load("passfinder/recomendations/service/models/kino.ann")
2023-05-27 11:13:41 +03:00
2023-05-28 00:11:44 +03:00
plays_model = AnnoyIndex(N_DIMENSIONAL, "angular")
plays_model.load("passfinder/recomendations/service/models/spektatli.ann")
2023-05-27 11:13:41 +03:00
2023-05-28 00:11:44 +03:00
excursion_model = AnnoyIndex(N_DIMENSIONAL, "angular")
excursion_model.load("passfinder/recomendations/service/models/excursii.ann")
2023-05-27 11:13:41 +03:00
2023-05-28 00:11:44 +03:00
concert_model = AnnoyIndex(N_DIMENSIONAL, "angular")
concert_model.load("passfinder/recomendations/service/models/concerts.ann")
2023-05-27 11:13:41 +03:00
2023-05-28 00:11:44 +03:00
mus_model = AnnoyIndex(N_DIMENSIONAL, "angular")
mus_model.load("passfinder/recomendations/service/models/mus.ann")