32 lines
738 B
Python
32 lines
738 B
Python
import implicit
|
|
|
|
from .mf_implicit_model import MFImplicitModel
|
|
|
|
|
|
class ALS(MFImplicitModel):
|
|
def __init__(
|
|
self,
|
|
latent_dim,
|
|
reg_term,
|
|
epochs,
|
|
random_state=42,
|
|
num_users=None,
|
|
num_items=None,
|
|
**kwargs,
|
|
):
|
|
super(ALS, self).__init__(
|
|
latent_dim=latent_dim,
|
|
reg_term=reg_term,
|
|
epochs=epochs,
|
|
learning_rate=None,
|
|
num_users=num_users,
|
|
num_items=num_items,
|
|
)
|
|
|
|
self.model = implicit.als.AlternatingLeastSquares(
|
|
factors=self.latent_dim,
|
|
regularization=self.reg_term,
|
|
iterations=self.epochs,
|
|
random_state=random_state,
|
|
)
|