FlexMF WARP

This page analyzes the hyperparameter tuning results for the FlexMF scorer in implicit-feedback mode with WARP loss.

Parameter Search Space

Parameter	Type	Distribution	Values	Selected
embedding_size	Integer	LogUniform	4 ≤ \(x\) ≤ 512	38
regularization	Float	LogUniform	0.0001 ≤ \(x\) ≤ 10	0.0633
learning_rate	Float	LogUniform	0.001 ≤ \(x\) ≤ 0.1	0.0024
reg_method	Categorical	Uniform	L2, AdamW	AdamW
item_bias	Categorical	Uniform	True, False	True

Final Result

Searching selected the following configuration:

{
    'embedding_size': 38,
    'regularization': 0.06330491699634001,
    'learning_rate': 0.0024039560315055057,
    'reg_method': 'AdamW',
    'item_bias': True,
    'epochs': 16
}

With these metrics:

{
    'RBP': 0.2262860366867493,
    'NDCG': 0.4446991199926725,
    'RecipRank': 0.4096242004925731,
    'TrainTask': 'b5b86e17-b137-4b14-8518-b3c6afc42312',
    'TrainTime': None,
    'TrainCPU': None,
    'max_epochs': 50,
    'done': False,
    'training_iteration': 16,
    'trial_id': 'b63fc691',
    'date': '2025-05-02_14-00-28',
    'timestamp': 1746208828,
    'time_this_iter_s': 53.725589990615845,
    'time_total_s': 891.8390142917633,
    'pid': 861825,
    'hostname': 'CCI-ws21',
    'node_ip': '10.248.127.152',
    'config': {
        'embedding_size': 38,
        'regularization': 0.06330491699634001,
        'learning_rate': 0.0024039560315055057,
        'reg_method': 'AdamW',
        'item_bias': True,
        'epochs': 16
    },
    'time_since_restore': 891.8390142917633,
    'iterations_since_restore': 16
}

Parameter Analysis

Embedding Size

The embedding size is the hyperparameter that most affects the model’s fundamental logic, so let’s look at performance as a fufnction of it:

Learning Parameters

Learning Rate

Regularization

Iteration Completion

How many iterations, on average, did we complete?

How did the metric progress in the best result?

How did the metric progress in the longest results?