ethen8181

import numpy as np
from lightfm import LightFM
from sklearn.metrics import roc_auc_score
from lightfm.datasets import fetch_movielens

def auc_score(model, ratings):
    """
    computes area under the ROC curve (AUC).
    The full name should probably be mean
    auc score as it is computing the auc

ethen8181

ethen8181

ethen8181

# MxN matrix A and N sized vector b. Ab = 

# [1, 0, 3]
# [0, 0, 0]
# [0, 2, 4]

# [1, 2, 3]

import numpy as np
cols = np.array([0, 2, 1, 2])