def load_rest():
User.load_to_cache()
# Load main models
Popularity.load_to_cache()
TensorCoFi.load_to_cache()
if "recommendation.language" in settings.INSTALLED_APPS:
from recommendation.language.models import Region
Region.load_to_cache()
if "recommendation.diversity" in settings.INSTALLED_APPS:
from recommendation.diversity.models import ItemGenre, Genre
Genre.load_to_cache()
ItemGenre.load_to_cache()
def setup_class(cls, *args, **kwargs):
"""
Put elements in db
"""
for app in ITEMS:
Item.objects.create(**app)
for u in USERS:
user = User.objects.create(external_id=u["external_id"])
for i in u["items"]:
Inventory.objects.create(user=user, item=Item.get_item_by_external_id(i))
TensorCoFi.train_from_db()
Popularity.train_from_db()
TensorCoFi.load_to_cache()
Popularity.load_to_cache()
def setup_class(cls, *args, **kwargs):
"""
Put elements in db
"""
path = resource_filename(recommendation.__name__, "/")
fill.FillTool({"items": True, "--mozilla": True, "prod": True}).load()
fill.FillTool({"users": True, "--mozilla": True, "<path>": path+"data/user"}).load()
modelcrafter.main("train", "popularity")
modelcrafter.main("train", "tensorcofi")
# Load user and items
Item.load_to_cache()
User.load_to_cache()
# Load main models
Popularity.load_to_cache()
TensorCoFi.load_to_cache()
cls.client = Client()
def test_training(self):
"""
[recommendation.models.TensorCoFi] Test train from database
"""
try:
TensorCoFi.train_from_db()
except Exception:
assert False, "Training is not working for jumping ids"
TensorCoFi.load_to_cache()
t = TensorCoFi.get_model_from_cache()
for user in User.objects.all():
if len(user.owned_items) > 2:
assert isinstance(t.get_recommendation(user), np.ndarray), "Recommendation is not a numpy array"
else:
try:
t.get_recommendation(user)
except KeyError:
pass
else:
assert False, "User with less than 3 items give a static recommendation"
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "recommendation.default_settings")
from django.core.wsgi import get_wsgi_application
application = get_wsgi_application()
from django.conf import settings
from recommendation.models import Item, User, TensorCoFi, Popularity
# Load user and items
Item.load_to_cache()
User.load_to_cache()
# Load main models
Popularity.load_to_cache()
TensorCoFi.load_to_cache()
if "recommendation.language" in settings.INSTALLED_APPS:
from recommendation.language.models import Locale, Region
Locale.load_to_cache()
Region.load_to_cache()
#if "recommendation.simple_logging" in recommendation.settings.INSTALLED_APPS:
# print("Loading logs to cache...")
# from recommendation.simple_logging.models import LogEntry
# LogEntry.load_to_cache()
# print("done!")
if "recommendation.diversity" in settings.INSTALLED_APPS:
"recommendation.default_settings")
from django.core.wsgi import get_wsgi_application
application = get_wsgi_application()
from django.conf import settings
from recommendation.models import Item, User, TensorCoFi, Popularity
# Load user and items
Item.load_to_cache()
User.load_to_cache()
# Load main models
Popularity.load_to_cache()
TensorCoFi.load_to_cache()
if "recommendation.language" in settings.INSTALLED_APPS:
from recommendation.language.models import Locale, Region
Locale.load_to_cache()
Region.load_to_cache()
#if "recommendation.simple_logging" in recommendation.settings.INSTALLED_APPS:
# print("Loading logs to cache...")
# from recommendation.simple_logging.models import LogEntry
# LogEntry.load_to_cache()
# print("done!")
if "recommendation.diversity" in settings.INSTALLED_APPS:
from recommendation.diversity.models import ItemGenre, Genre
Genre.load_to_cache()
