Ejemplo n.º 1
0
def test_all_hyperparameters(sagemaker_session):
    fm = FactorizationMachines(
        sagemaker_session=sagemaker_session,
        epochs=2,
        clip_gradient=1e2,
        eps=0.001,
        rescale_grad=2.2,
        bias_lr=0.01,
        linear_lr=0.002,
        factors_lr=0.0003,
        bias_wd=0.0004,
        linear_wd=1.01,
        factors_wd=1.002,
        bias_init_method="uniform",
        bias_init_scale=0.1,
        bias_init_sigma=0.05,
        bias_init_value=2.002,
        linear_init_method="constant",
        linear_init_scale=0.02,
        linear_init_sigma=0.003,
        linear_init_value=1.0,
        factors_init_method="normal",
        factors_init_scale=1.101,
        factors_init_sigma=1.202,
        factors_init_value=1.303,
        **ALL_REQ_ARGS,
    )
    assert fm.hyperparameters() == dict(
        num_factors=str(ALL_REQ_ARGS["num_factors"]),
        predictor_type=ALL_REQ_ARGS["predictor_type"],
        epochs="2",
        clip_gradient="100.0",
        eps="0.001",
        rescale_grad="2.2",
        bias_lr="0.01",
        linear_lr="0.002",
        factors_lr="0.0003",
        bias_wd="0.0004",
        linear_wd="1.01",
        factors_wd="1.002",
        bias_init_method="uniform",
        bias_init_scale="0.1",
        bias_init_sigma="0.05",
        bias_init_value="2.002",
        linear_init_method="constant",
        linear_init_scale="0.02",
        linear_init_sigma="0.003",
        linear_init_value="1.0",
        factors_init_method="normal",
        factors_init_scale="1.101",
        factors_init_sigma="1.202",
        factors_init_value="1.303",
    )
Ejemplo n.º 2
0
def test_all_hyperparameters(sagemaker_session):
    fm = FactorizationMachines(sagemaker_session=sagemaker_session,
                               epochs=2,
                               clip_gradient=1e2,
                               eps=0.001,
                               rescale_grad=2.2,
                               bias_lr=0.01,
                               linear_lr=0.002,
                               factors_lr=0.0003,
                               bias_wd=0.0004,
                               linear_wd=1.01,
                               factors_wd=1.002,
                               bias_init_method='uniform',
                               bias_init_scale=0.1,
                               bias_init_sigma=0.05,
                               bias_init_value=2.002,
                               linear_init_method='constant',
                               linear_init_scale=0.02,
                               linear_init_sigma=0.003,
                               linear_init_value=1.0,
                               factors_init_method='normal',
                               factors_init_scale=1.101,
                               factors_init_sigma=1.202,
                               factors_init_value=1.303,
                               **ALL_REQ_ARGS)
    assert fm.hyperparameters() == dict(
        num_factors=str(ALL_REQ_ARGS['num_factors']),
        predictor_type=ALL_REQ_ARGS['predictor_type'],
        epochs='2',
        clip_gradient='100.0',
        eps='0.001',
        rescale_grad='2.2',
        bias_lr='0.01',
        linear_lr='0.002',
        factors_lr='0.0003',
        bias_wd='0.0004',
        linear_wd='1.01',
        factors_wd='1.002',
        bias_init_method='uniform',
        bias_init_scale='0.1',
        bias_init_sigma='0.05',
        bias_init_value='2.002',
        linear_init_method='constant',
        linear_init_scale='0.02',
        linear_init_sigma='0.003',
        linear_init_value='1.0',
        factors_init_method='normal',
        factors_init_scale='1.101',
        factors_init_sigma='1.202',
        factors_init_value='1.303',
    )
Ejemplo n.º 3
0
def test_all_hyperparameters(sagemaker_session):
    fm = FactorizationMachines(sagemaker_session=sagemaker_session,
                               epochs=2, clip_gradient=1e2, eps=0.001, rescale_grad=2.2,
                               bias_lr=0.01, linear_lr=0.002, factors_lr=0.0003,
                               bias_wd=0.0004, linear_wd=1.01, factors_wd=1.002,
                               bias_init_method='uniform', bias_init_scale=0.1, bias_init_sigma=0.05,
                               bias_init_value=2.002, linear_init_method='constant', linear_init_scale=0.02,
                               linear_init_sigma=0.003, linear_init_value=1.0, factors_init_method='normal',
                               factors_init_scale=1.101, factors_init_sigma=1.202, factors_init_value=1.303,
                               **ALL_REQ_ARGS)
    assert fm.hyperparameters() == dict(
        num_factors=str(ALL_REQ_ARGS['num_factors']),
        predictor_type=ALL_REQ_ARGS['predictor_type'],
        epochs='2',
        clip_gradient='100.0',
        eps='0.001',
        rescale_grad='2.2',
        bias_lr='0.01',
        linear_lr='0.002',
        factors_lr='0.0003',
        bias_wd='0.0004',
        linear_wd='1.01',
        factors_wd='1.002',
        bias_init_method='uniform',
        bias_init_scale='0.1',
        bias_init_sigma='0.05',
        bias_init_value='2.002',
        linear_init_method='constant',
        linear_init_scale='0.02',
        linear_init_sigma='0.003',
        linear_init_value='1.0',
        factors_init_method='normal',
        factors_init_scale='1.101',
        factors_init_sigma='1.202',
        factors_init_value='1.303',
    )