Commit 7be7cfe2 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented weight saving callback in model_training.py

parent f851531e
......@@ -384,8 +384,8 @@ else:
generator,
grouper,
gmvaep,
mmd_warmup_callback,
kl_warmup_callback,
mmd_warmup_callback,
) = SEQ_2_SEQ_GMVAE(
input_dict_train[input_type].shape,
loss=loss,
......
......@@ -352,9 +352,7 @@ dfcats_max = pd.concat(
[
pd.DataFrame(
(
[str(int(i)) for i in np.random.uniform(0, k, samples)]
if variational
else np.zeros(samples)
np.zeros(samples)
)
),
pd.DataFrame(np.array(np.max((np.concatenate(clusters)), axis=1), dtype=str)),
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment