def test_all_hyperparameters(sagemaker_session):
    lda = LDA(sagemaker_session=sagemaker_session,
              alpha0=2.2, max_restarts=3, max_iterations=10, tol=3.3,
              **ALL_REQ_ARGS)
    assert lda.hyperparameters() == dict(
        num_topics=str(ALL_REQ_ARGS['num_topics']),
        alpha0='2.2',
        max_restarts='3',
        max_iterations='10',
        tol='3.3',
    )
Example #2
0
def test_all_hyperparameters(sagemaker_session):
    lda = LDA(sagemaker_session=sagemaker_session,
              alpha0=2.2,
              max_restarts=3,
              max_iterations=10,
              tol=3.3,
              **ALL_REQ_ARGS)
    assert lda.hyperparameters() == dict(
        num_topics=str(ALL_REQ_ARGS['num_topics']),
        alpha0='2.2',
        max_restarts='3',
        max_iterations='10',
        tol='3.3',
    )
def test_all_hyperparameters(sagemaker_session):
    lda = LDA(sagemaker_session=sagemaker_session,
              alpha0=2.2,
              max_restarts=3,
              max_iterations=10,
              tol=3.3,
              **ALL_REQ_ARGS)
    assert lda.hyperparameters() == dict(
        num_topics=str(ALL_REQ_ARGS["num_topics"]),
        alpha0="2.2",
        max_restarts="3",
        max_iterations="10",
        tol="3.3",
    )