def test_all_hyperparameters(sagemaker_session):
kmeans = KMeans(sagemaker_session=sagemaker_session,
init_method='random',
max_iterations=3,
tol=0.5,
num_trials=5,
local_init_method='kmeans++',
half_life_time_size=0,
epochs=10,
center_factor=2,
eval_metrics=['msd', 'ssd'],
**ALL_REQ_ARGS)
assert kmeans.hyperparameters() == dict(
k=str(ALL_REQ_ARGS['k']),
init_method='random',
local_lloyd_max_iter='3',
local_lloyd_tol='0.5',
local_lloyd_num_trials='5',
local_lloyd_init_method='kmeans++',
half_life_time_size='0',
epochs='10',
extra_center_factor='2',
eval_metrics='[\'msd\', \'ssd\']',
force_dense='True',
)
def test_all_hyperparameters(sagemaker_session):
kmeans = KMeans(sagemaker_session=sagemaker_session,
init_method="random",
max_iterations=3,
tol=0.5,
num_trials=5,
local_init_method="kmeans++",
half_life_time_size=0,
epochs=10,
center_factor=2,
eval_metrics=["msd", "ssd"],
**ALL_REQ_ARGS)
assert kmeans.hyperparameters() == dict(
k=str(ALL_REQ_ARGS["k"]),
init_method="random",
local_lloyd_max_iter="3",
local_lloyd_tol="0.5",
local_lloyd_num_trials="5",
local_lloyd_init_method="kmeans++",
half_life_time_size="0",
epochs="10",
extra_center_factor="2",
eval_metrics='["msd", "ssd"]',
force_dense="True",
)
def test_all_hyperparameters(sagemaker_session):
kmeans = KMeans(sagemaker_session=sagemaker_session, init_method='random', max_iterations=3, tol=0.5,
num_trials=5, local_init_method='kmeans++', half_life_time_size=0, epochs=10, center_factor=2,
eval_metrics=['msd', 'ssd'], **ALL_REQ_ARGS)
assert kmeans.hyperparameters() == dict(
k=str(ALL_REQ_ARGS['k']),
init_method='random',
local_lloyd_max_iter='3',
local_lloyd_tol='0.5',
local_lloyd_num_trials='5',
local_lloyd_init_method='kmeans++',
half_life_time_size='0',
epochs='10',
extra_center_factor='2',
eval_metrics='[\'msd\', \'ssd\']',
force_dense='True',
)
