import scanpy as sc
import sys
sys.path.append("../")
import trvaep
import numpy as np
adata = sc.read("../data/haber_count.h5ad", backup_url="shorturl.at/imuEP")
sc.pp.normalize_per_cell(adata)
sc.pp.log1p(adata)
sc.pp.highly_variable_genes(adata, n_top_genes=2000)
adata = adata[:, adata.var['highly_variable']]
n_conditions = adata.obs["condition"].unique().shape[0]
condition_key = "condition"
adata_train = adata[~((adata.obs["cell_label"] == "TA")
& (adata.obs["condition"] == "Hpoly.Day10"))]
sc.pp.neighbors(adata_train)
sc.tl.umap(adata_train)
WARNING: You’re trying to run this on 2000 dimensions of `.X`, if you really want this, set `use_rep='X'`. Falling back to preprocessing with `sc.pp.pca` and default params.
sc.pl.umap(adata_train, color=["condition", "cell_label"], wspace=.4)
model = trvaep.CVAE(adata_train.n_vars, num_classes=n_conditions,
output_activation="ReLU", encoder_layer_sizes=[64, 32], decoder_layer_sizes=[32, 64], use_mmd=True, beta=1)
trainer = trvaep.Trainer(model, adata_train, condition_key=condition_key)
trainer.train_trvae(n_epochs=300, batch_size=512, early_patience=50)
Epoch 00/300 Batch 0000/15, Loss: 388.9540, rec_loss: 387.3120, KL_loss: 0.0021, MMD_loss: 1.6400 Epoch 00/300 Batch 0015/15, Loss: 233.0981, rec_loss: 231.4851, KL_loss: 0.0028, MMD_loss: 1.6102 Epoch 01/300 Batch 0000/15, Loss: 226.2108, rec_loss: 224.6947, KL_loss: 0.0029, MMD_loss: 1.5131 Epoch 01/300 Batch 0015/15, Loss: 184.3745, rec_loss: 183.0172, KL_loss: 0.0040, MMD_loss: 1.3533 Epoch 02/300 Batch 0000/15, Loss: 187.5044, rec_loss: 186.1422, KL_loss: 0.0040, MMD_loss: 1.3582 Epoch 02/300 Batch 0015/15, Loss: 160.1572, rec_loss: 158.7332, KL_loss: 0.0048, MMD_loss: 1.4192 Epoch 03/300 Batch 0000/15, Loss: 161.5090, rec_loss: 160.1668, KL_loss: 0.0049, MMD_loss: 1.3373 Epoch 03/300 Batch 0015/15, Loss: 149.5970, rec_loss: 148.3430, KL_loss: 0.0065, MMD_loss: 1.2475 Epoch 04/300 Batch 0000/15, Loss: 146.4888, rec_loss: 145.2492, KL_loss: 0.0065, MMD_loss: 1.2331 Epoch 04/300 Batch 0015/15, Loss: 133.3064, rec_loss: 131.8642, KL_loss: 0.0082, MMD_loss: 1.4341 Epoch 05/300 Batch 0000/15, Loss: 132.8760, rec_loss: 131.6510, KL_loss: 0.0083, MMD_loss: 1.2167 Epoch 05/300 Batch 0015/15, Loss: 125.9368, rec_loss: 124.7598, KL_loss: 0.0109, MMD_loss: 1.1662 Epoch 06/300 Batch 0000/15, Loss: 125.5383, rec_loss: 124.4178, KL_loss: 0.0106, MMD_loss: 1.1099 Epoch 06/300 Batch 0015/15, Loss: 125.3534, rec_loss: 124.2306, KL_loss: 0.0122, MMD_loss: 1.1106 Epoch 07/300 Batch 0000/15, Loss: 118.5388, rec_loss: 117.2885, KL_loss: 0.0127, MMD_loss: 1.2375 Epoch 07/300 Batch 0015/15, Loss: 121.8226, rec_loss: 120.8375, KL_loss: 0.0140, MMD_loss: 0.9711 Epoch 08/300 Batch 0000/15, Loss: 117.2280, rec_loss: 116.1856, KL_loss: 0.0145, MMD_loss: 1.0278 Epoch 08/300 Batch 0015/15, Loss: 115.1225, rec_loss: 114.1035, KL_loss: 0.0173, MMD_loss: 1.0017 Epoch 09/300 Batch 0000/15, Loss: 113.4403, rec_loss: 112.2518, KL_loss: 0.0162, MMD_loss: 1.1723 Epoch 09/300 Batch 0015/15, Loss: 115.6096, rec_loss: 114.5371, KL_loss: 0.0175, MMD_loss: 1.0551 Epoch 10/300 Batch 0000/15, Loss: 117.8211, rec_loss: 116.6365, KL_loss: 0.0182, MMD_loss: 1.1664 Epoch 10/300 Batch 0015/15, Loss: 109.3953, rec_loss: 108.1537, KL_loss: 0.0195, MMD_loss: 1.2221 Epoch 11/300 Batch 0000/15, Loss: 113.9833, rec_loss: 112.9695, KL_loss: 0.0199, MMD_loss: 0.9939 Epoch 11/300 Batch 0015/15, Loss: 109.3150, rec_loss: 108.2208, KL_loss: 0.0215, MMD_loss: 1.0727 Epoch 12/300 Batch 0000/15, Loss: 110.5718, rec_loss: 109.4711, KL_loss: 0.0202, MMD_loss: 1.0805 Epoch 12/300 Batch 0015/15, Loss: 107.6259, rec_loss: 106.6547, KL_loss: 0.0220, MMD_loss: 0.9492 Epoch 13/300 Batch 0000/15, Loss: 106.3605, rec_loss: 105.2821, KL_loss: 0.0225, MMD_loss: 1.0559 Epoch 13/300 Batch 0015/15, Loss: 107.3077, rec_loss: 106.2209, KL_loss: 0.0236, MMD_loss: 1.0633 Epoch 14/300 Batch 0000/15, Loss: 112.5205, rec_loss: 111.4908, KL_loss: 0.0240, MMD_loss: 1.0056 Epoch 14/300 Batch 0015/15, Loss: 104.3965, rec_loss: 103.3148, KL_loss: 0.0253, MMD_loss: 1.0565 Epoch 15/300 Batch 0000/15, Loss: 108.7772, rec_loss: 107.7707, KL_loss: 0.0254, MMD_loss: 0.9812 Epoch 15/300 Batch 0015/15, Loss: 103.8460, rec_loss: 102.8916, KL_loss: 0.0269, MMD_loss: 0.9276 Epoch 16/300 Batch 0000/15, Loss: 103.8789, rec_loss: 102.8998, KL_loss: 0.0266, MMD_loss: 0.9524 Epoch 16/300 Batch 0015/15, Loss: 103.7652, rec_loss: 102.7331, KL_loss: 0.0283, MMD_loss: 1.0038 Epoch 17/300 Batch 0000/15, Loss: 102.8496, rec_loss: 101.9009, KL_loss: 0.0285, MMD_loss: 0.9202 Epoch 17/300 Batch 0015/15, Loss: 103.7155, rec_loss: 102.6402, KL_loss: 0.0292, MMD_loss: 1.0461 Epoch 18/300 Batch 0000/15, Loss: 101.7910, rec_loss: 100.8737, KL_loss: 0.0296, MMD_loss: 0.8877 Epoch 18/300 Batch 0015/15, Loss: 104.6859, rec_loss: 103.9105, KL_loss: 0.0298, MMD_loss: 0.7455 Epoch 19/300 Batch 0000/15, Loss: 103.0755, rec_loss: 101.9458, KL_loss: 0.0315, MMD_loss: 1.0982 Epoch 19/300 Batch 0015/15, Loss: 100.1365, rec_loss: 99.3377, KL_loss: 0.0333, MMD_loss: 0.7655 Epoch 20/300 Batch 0000/15, Loss: 103.7283, rec_loss: 102.8521, KL_loss: 0.0314, MMD_loss: 0.8448 Epoch 20/300 Batch 0015/15, Loss: 99.6207, rec_loss: 98.7867, KL_loss: 0.0345, MMD_loss: 0.7996 Epoch 20, Loss_valid: 152.4705, rec_loss_valid: 138.7722, KL_loss_valid: 0.0475, MMD_loss: 8.5716 Epoch 21/300 Batch 0000/15, Loss: 97.9728, rec_loss: 97.1486, KL_loss: 0.0338, MMD_loss: 0.7904 Epoch 21/300 Batch 0015/15, Loss: 101.8258, rec_loss: 100.9972, KL_loss: 0.0347, MMD_loss: 0.7939 Epoch 22/300 Batch 0000/15, Loss: 101.0762, rec_loss: 100.2285, KL_loss: 0.0335, MMD_loss: 0.8142 Epoch 22/300 Batch 0015/15, Loss: 100.8727, rec_loss: 100.1056, KL_loss: 0.0354, MMD_loss: 0.7316 Epoch 23/300 Batch 0000/15, Loss: 99.9489, rec_loss: 99.0750, KL_loss: 0.0351, MMD_loss: 0.8389 Epoch 23/300 Batch 0015/15, Loss: 102.9383, rec_loss: 102.1832, KL_loss: 0.0364, MMD_loss: 0.7187 Epoch 24/300 Batch 0000/15, Loss: 98.4898, rec_loss: 97.6806, KL_loss: 0.0377, MMD_loss: 0.7715 Epoch 24/300 Batch 0015/15, Loss: 99.8968, rec_loss: 99.0205, KL_loss: 0.0387, MMD_loss: 0.8376 Epoch 25/300 Batch 0000/15, Loss: 98.6542, rec_loss: 97.8880, KL_loss: 0.0386, MMD_loss: 0.7276 Epoch 25/300 Batch 0015/15, Loss: 101.1164, rec_loss: 100.3963, KL_loss: 0.0391, MMD_loss: 0.6809 Epoch 26/300 Batch 0000/15, Loss: 99.0320, rec_loss: 98.2326, KL_loss: 0.0398, MMD_loss: 0.7597 Epoch 26/300 Batch 0015/15, Loss: 99.2768, rec_loss: 98.5863, KL_loss: 0.0413, MMD_loss: 0.6492 Epoch 27/300 Batch 0000/15, Loss: 97.5860, rec_loss: 96.8916, KL_loss: 0.0408, MMD_loss: 0.6536 Epoch 27/300 Batch 0015/15, Loss: 102.5444, rec_loss: 101.9310, KL_loss: 0.0414, MMD_loss: 0.5719 Epoch 28/300 Batch 0000/15, Loss: 102.3181, rec_loss: 101.5902, KL_loss: 0.0412, MMD_loss: 0.6866 Epoch 28/300 Batch 0015/15, Loss: 96.3414, rec_loss: 95.6174, KL_loss: 0.0416, MMD_loss: 0.6824 Epoch 29/300 Batch 0000/15, Loss: 97.2571, rec_loss: 96.5895, KL_loss: 0.0422, MMD_loss: 0.6254 Epoch 29/300 Batch 0015/15, Loss: 101.9391, rec_loss: 101.2105, KL_loss: 0.0424, MMD_loss: 0.6862 Epoch 30/300 Batch 0000/15, Loss: 96.9122, rec_loss: 96.2445, KL_loss: 0.0443, MMD_loss: 0.6235 Epoch 30/300 Batch 0015/15, Loss: 96.2947, rec_loss: 95.6894, KL_loss: 0.0436, MMD_loss: 0.5617 Epoch 31/300 Batch 0000/15, Loss: 98.8109, rec_loss: 98.0892, KL_loss: 0.0456, MMD_loss: 0.6761 Epoch 31/300 Batch 0015/15, Loss: 96.0713, rec_loss: 95.4416, KL_loss: 0.0462, MMD_loss: 0.5835 Epoch 32/300 Batch 0000/15, Loss: 95.7035, rec_loss: 95.0565, KL_loss: 0.0459, MMD_loss: 0.6012 Epoch 32/300 Batch 0015/15, Loss: 100.8639, rec_loss: 100.2247, KL_loss: 0.0454, MMD_loss: 0.5938 Epoch 33/300 Batch 0000/15, Loss: 94.8681, rec_loss: 94.1860, KL_loss: 0.0470, MMD_loss: 0.6350 Epoch 33/300 Batch 0015/15, Loss: 99.4269, rec_loss: 98.7502, KL_loss: 0.0485, MMD_loss: 0.6282 Epoch 34/300 Batch 0000/15, Loss: 96.2527, rec_loss: 95.6205, KL_loss: 0.0480, MMD_loss: 0.5841 Epoch 34/300 Batch 0015/15, Loss: 96.0572, rec_loss: 95.3946, KL_loss: 0.0489, MMD_loss: 0.6137 Epoch 35/300 Batch 0000/15, Loss: 97.6811, rec_loss: 97.1085, KL_loss: 0.0491, MMD_loss: 0.5235 Epoch 35/300 Batch 0015/15, Loss: 95.5990, rec_loss: 95.0248, KL_loss: 0.0500, MMD_loss: 0.5242 Epoch 36/300 Batch 0000/15, Loss: 95.2522, rec_loss: 94.6580, KL_loss: 0.0508, MMD_loss: 0.5434 Epoch 36/300 Batch 0015/15, Loss: 96.5723, rec_loss: 95.8620, KL_loss: 0.0522, MMD_loss: 0.6582 Epoch 37/300 Batch 0000/15, Loss: 98.8443, rec_loss: 98.2113, KL_loss: 0.0503, MMD_loss: 0.5826 Epoch 37/300 Batch 0015/15, Loss: 96.1596, rec_loss: 95.6658, KL_loss: 0.0527, MMD_loss: 0.4411 Epoch 38/300 Batch 0000/15, Loss: 94.9612, rec_loss: 94.2686, KL_loss: 0.0517, MMD_loss: 0.6409 Epoch 38/300 Batch 0015/15, Loss: 98.5880, rec_loss: 98.0014, KL_loss: 0.0534, MMD_loss: 0.5332 Epoch 39/300 Batch 0000/15, Loss: 96.9878, rec_loss: 96.3081, KL_loss: 0.0519, MMD_loss: 0.6277 Epoch 39/300 Batch 0015/15, Loss: 93.0073, rec_loss: 92.5086, KL_loss: 0.0544, MMD_loss: 0.4443 Epoch 40/300 Batch 0000/15, Loss: 91.6633, rec_loss: 91.0413, KL_loss: 0.0542, MMD_loss: 0.5678 Epoch 40/300 Batch 0015/15, Loss: 92.0270, rec_loss: 91.4154, KL_loss: 0.0567, MMD_loss: 0.5550 Epoch 40, Loss_valid: 139.9258, rec_loss_valid: 129.7552, KL_loss_valid: 0.0767, MMD_loss: 7.1165 Epoch 41/300 Batch 0000/15, Loss: 95.7662, rec_loss: 95.1298, KL_loss: 0.0534, MMD_loss: 0.5830 Epoch 41/300 Batch 0015/15, Loss: 96.2324, rec_loss: 95.6190, KL_loss: 0.0560, MMD_loss: 0.5573 Epoch 42/300 Batch 0000/15, Loss: 97.5008, rec_loss: 96.9811, KL_loss: 0.0576, MMD_loss: 0.4621 Epoch 42/300 Batch 0015/15, Loss: 99.2509, rec_loss: 98.6990, KL_loss: 0.0571, MMD_loss: 0.4948 Epoch 43/300 Batch 0000/15, Loss: 94.9210, rec_loss: 94.3867, KL_loss: 0.0584, MMD_loss: 0.4758 Epoch 43/300 Batch 0015/15, Loss: 97.9498, rec_loss: 97.4230, KL_loss: 0.0589, MMD_loss: 0.4679 Epoch 44/300 Batch 0000/15, Loss: 93.4733, rec_loss: 92.8616, KL_loss: 0.0589, MMD_loss: 0.5528 Epoch 44/300 Batch 0015/15, Loss: 93.5125, rec_loss: 92.9836, KL_loss: 0.0601, MMD_loss: 0.4688 Epoch 45/300 Batch 0000/15, Loss: 92.7321, rec_loss: 92.2407, KL_loss: 0.0582, MMD_loss: 0.4332 Epoch 45/300 Batch 0015/15, Loss: 92.9617, rec_loss: 92.3764, KL_loss: 0.0604, MMD_loss: 0.5248 Epoch 46/300 Batch 0000/15, Loss: 92.3981, rec_loss: 91.8104, KL_loss: 0.0610, MMD_loss: 0.5268 Epoch 46/300 Batch 0015/15, Loss: 97.2236, rec_loss: 96.7346, KL_loss: 0.0637, MMD_loss: 0.4253 Epoch 47/300 Batch 0000/15, Loss: 94.5074, rec_loss: 93.9385, KL_loss: 0.0607, MMD_loss: 0.5082 Epoch 47/300 Batch 0015/15, Loss: 96.9680, rec_loss: 96.5450, KL_loss: 0.0620, MMD_loss: 0.3609 Epoch 48/300 Batch 0000/15, Loss: 89.6372, rec_loss: 89.1122, KL_loss: 0.0630, MMD_loss: 0.4620 Epoch 48/300 Batch 0015/15, Loss: 91.2100, rec_loss: 90.7677, KL_loss: 0.0645, MMD_loss: 0.3777 Epoch 49/300 Batch 0000/15, Loss: 92.8206, rec_loss: 92.2682, KL_loss: 0.0646, MMD_loss: 0.4878 Epoch 49/300 Batch 0015/15, Loss: 91.7414, rec_loss: 91.1560, KL_loss: 0.0640, MMD_loss: 0.5214 Epoch 50/300 Batch 0000/15, Loss: 94.1428, rec_loss: 93.6626, KL_loss: 0.0630, MMD_loss: 0.4172 Epoch 50/300 Batch 0015/15, Loss: 94.4778, rec_loss: 94.0603, KL_loss: 0.0653, MMD_loss: 0.3522 Epoch 51/300 Batch 0000/15, Loss: 93.6700, rec_loss: 93.1469, KL_loss: 0.0659, MMD_loss: 0.4572 Epoch 51/300 Batch 0015/15, Loss: 93.7240, rec_loss: 93.0918, KL_loss: 0.0674, MMD_loss: 0.5648 Epoch 52/300 Batch 0000/15, Loss: 93.0067, rec_loss: 92.4363, KL_loss: 0.0676, MMD_loss: 0.5028 Epoch 52/300 Batch 0015/15, Loss: 96.0216, rec_loss: 95.5818, KL_loss: 0.0682, MMD_loss: 0.3716 Epoch 53/300 Batch 0000/15, Loss: 91.7707, rec_loss: 91.2492, KL_loss: 0.0660, MMD_loss: 0.4555 Epoch 53/300 Batch 0015/15, Loss: 90.6525, rec_loss: 90.1831, KL_loss: 0.0679, MMD_loss: 0.4015 Epoch 54/300 Batch 0000/15, Loss: 94.1411, rec_loss: 93.6942, KL_loss: 0.0672, MMD_loss: 0.3798 Epoch 54/300 Batch 0015/15, Loss: 92.6665, rec_loss: 91.9768, KL_loss: 0.0689, MMD_loss: 0.6209 Epoch 55/300 Batch 0000/15, Loss: 91.8499, rec_loss: 91.3571, KL_loss: 0.0698, MMD_loss: 0.4231 Epoch 55/300 Batch 0015/15, Loss: 88.2741, rec_loss: 87.7710, KL_loss: 0.0691, MMD_loss: 0.4339 Epoch 56/300 Batch 0000/15, Loss: 94.5690, rec_loss: 94.0643, KL_loss: 0.0707, MMD_loss: 0.4340 Epoch 56/300 Batch 0015/15, Loss: 92.6941, rec_loss: 92.2198, KL_loss: 0.0711, MMD_loss: 0.4033 Epoch 57/300 Batch 0000/15, Loss: 93.8389, rec_loss: 93.2952, KL_loss: 0.0728, MMD_loss: 0.4709 Epoch 57/300 Batch 0015/15, Loss: 89.2589, rec_loss: 88.7744, KL_loss: 0.0713, MMD_loss: 0.4131 Epoch 58/300 Batch 0000/15, Loss: 93.0163, rec_loss: 92.4687, KL_loss: 0.0730, MMD_loss: 0.4746 Epoch 58/300 Batch 0015/15, Loss: 93.5987, rec_loss: 93.1535, KL_loss: 0.0733, MMD_loss: 0.3719 Epoch 59/300 Batch 0000/15, Loss: 92.7984, rec_loss: 92.3051, KL_loss: 0.0734, MMD_loss: 0.4199 Epoch 59/300 Batch 0015/15, Loss: 92.5970, rec_loss: 92.1633, KL_loss: 0.0740, MMD_loss: 0.3597 Epoch 60/300 Batch 0000/15, Loss: 92.8855, rec_loss: 92.4460, KL_loss: 0.0736, MMD_loss: 0.3659 Epoch 60/300 Batch 0015/15, Loss: 91.7478, rec_loss: 91.2812, KL_loss: 0.0742, MMD_loss: 0.3924 Epoch 60, Loss_valid: 134.1593, rec_loss_valid: 125.8761, KL_loss_valid: 0.1027, MMD_loss: 5.1215 Epoch 61/300 Batch 0000/15, Loss: 95.4177, rec_loss: 94.9260, KL_loss: 0.0755, MMD_loss: 0.4161 Epoch 61/300 Batch 0015/15, Loss: 92.2241, rec_loss: 91.7585, KL_loss: 0.0783, MMD_loss: 0.3872 Epoch 62/300 Batch 0000/15, Loss: 93.3021, rec_loss: 92.7413, KL_loss: 0.0752, MMD_loss: 0.4855 Epoch 62/300 Batch 0015/15, Loss: 94.4946, rec_loss: 93.9923, KL_loss: 0.0770, MMD_loss: 0.4253 Epoch 63/300 Batch 0000/15, Loss: 91.2771, rec_loss: 90.8681, KL_loss: 0.0766, MMD_loss: 0.3325 Epoch 63/300 Batch 0015/15, Loss: 91.2541, rec_loss: 90.8367, KL_loss: 0.0764, MMD_loss: 0.3411 Epoch 64/300 Batch 0000/15, Loss: 91.9053, rec_loss: 91.3075, KL_loss: 0.0784, MMD_loss: 0.5194 Epoch 64/300 Batch 0015/15, Loss: 89.3330, rec_loss: 88.9184, KL_loss: 0.0809, MMD_loss: 0.3336 Epoch 65/300 Batch 0000/15, Loss: 94.1233, rec_loss: 93.6512, KL_loss: 0.0778, MMD_loss: 0.3943 Epoch 65/300 Batch 0015/15, Loss: 93.6249, rec_loss: 93.2032, KL_loss: 0.0765, MMD_loss: 0.3451 Epoch 66/300 Batch 0000/15, Loss: 92.5460, rec_loss: 92.0680, KL_loss: 0.0782, MMD_loss: 0.3998 Epoch 66/300 Batch 0015/15, Loss: 92.7501, rec_loss: 92.3963, KL_loss: 0.0827, MMD_loss: 0.2710 Epoch 67/300 Batch 0000/15, Loss: 93.3277, rec_loss: 92.9298, KL_loss: 0.0800, MMD_loss: 0.3179 Epoch 67/300 Batch 0015/15, Loss: 89.2257, rec_loss: 88.8194, KL_loss: 0.0798, MMD_loss: 0.3265 Epoch 68/300 Batch 0000/15, Loss: 94.1738, rec_loss: 93.7309, KL_loss: 0.0794, MMD_loss: 0.3634 Epoch 68/300 Batch 0015/15, Loss: 92.5483, rec_loss: 92.1653, KL_loss: 0.0820, MMD_loss: 0.3010 Epoch 69/300 Batch 0000/15, Loss: 90.9699, rec_loss: 90.5390, KL_loss: 0.0839, MMD_loss: 0.3471 Epoch 69/300 Batch 0015/15, Loss: 92.4141, rec_loss: 91.9094, KL_loss: 0.0800, MMD_loss: 0.4247 Epoch 70/300 Batch 0000/15, Loss: 91.0081, rec_loss: 90.5195, KL_loss: 0.0830, MMD_loss: 0.4057 Epoch 70/300 Batch 0015/15, Loss: 96.0755, rec_loss: 95.5812, KL_loss: 0.0822, MMD_loss: 0.4121 Epoch 71/300 Batch 0000/15, Loss: 90.4734, rec_loss: 90.0666, KL_loss: 0.0815, MMD_loss: 0.3253 Epoch 71/300 Batch 0015/15, Loss: 93.1048, rec_loss: 92.6057, KL_loss: 0.0831, MMD_loss: 0.4159 Epoch 72/300 Batch 0000/15, Loss: 96.1522, rec_loss: 95.7036, KL_loss: 0.0839, MMD_loss: 0.3647 Epoch 72/300 Batch 0015/15, Loss: 93.4172, rec_loss: 92.9925, KL_loss: 0.0856, MMD_loss: 0.3391 Epoch 73/300 Batch 0000/15, Loss: 89.3970, rec_loss: 88.9511, KL_loss: 0.0846, MMD_loss: 0.3614 Epoch 73/300 Batch 0015/15, Loss: 93.7914, rec_loss: 93.3258, KL_loss: 0.0843, MMD_loss: 0.3813 Epoch 74/300 Batch 0000/15, Loss: 89.6929, rec_loss: 89.2347, KL_loss: 0.0874, MMD_loss: 0.3708 Epoch 74/300 Batch 0015/15, Loss: 89.9258, rec_loss: 89.4855, KL_loss: 0.0856, MMD_loss: 0.3547 Epoch 75/300 Batch 0000/15, Loss: 89.8210, rec_loss: 89.4416, KL_loss: 0.0880, MMD_loss: 0.2914 Epoch 75/300 Batch 0015/15, Loss: 94.1168, rec_loss: 93.7086, KL_loss: 0.0871, MMD_loss: 0.3211 Epoch 76/300 Batch 0000/15, Loss: 91.2257, rec_loss: 90.8037, KL_loss: 0.0885, MMD_loss: 0.3335 Epoch 76/300 Batch 0015/15, Loss: 91.9996, rec_loss: 91.4150, KL_loss: 0.0862, MMD_loss: 0.4984 Epoch 77/300 Batch 0000/15, Loss: 91.5255, rec_loss: 91.1113, KL_loss: 0.0881, MMD_loss: 0.3261 Epoch 77/300 Batch 0015/15, Loss: 90.5517, rec_loss: 90.1634, KL_loss: 0.0874, MMD_loss: 0.3009 Epoch 78/300 Batch 0000/15, Loss: 88.9422, rec_loss: 88.5532, KL_loss: 0.0877, MMD_loss: 0.3012 Epoch 78/300 Batch 0015/15, Loss: 95.0782, rec_loss: 94.6333, KL_loss: 0.0872, MMD_loss: 0.3577 Epoch 79/300 Batch 0000/15, Loss: 87.9678, rec_loss: 87.5572, KL_loss: 0.0945, MMD_loss: 0.3162 Epoch 79/300 Batch 0015/15, Loss: 91.1911, rec_loss: 90.7658, KL_loss: 0.0912, MMD_loss: 0.3340 Epoch 80/300 Batch 0000/15, Loss: 91.8428, rec_loss: 91.4381, KL_loss: 0.0910, MMD_loss: 0.3138 Epoch 80/300 Batch 0015/15, Loss: 89.3939, rec_loss: 89.0102, KL_loss: 0.0918, MMD_loss: 0.2919 Epoch 80, Loss_valid: 131.1122, rec_loss_valid: 124.2599, KL_loss_valid: 0.1255, MMD_loss: 5.0990 Epoch 81/300 Batch 0000/15, Loss: 91.5512, rec_loss: 91.0955, KL_loss: 0.0911, MMD_loss: 0.3646 Epoch 81/300 Batch 0015/15, Loss: 89.0811, rec_loss: 88.6714, KL_loss: 0.0937, MMD_loss: 0.3160 Epoch 82/300 Batch 0000/15, Loss: 91.6422, rec_loss: 91.2143, KL_loss: 0.0917, MMD_loss: 0.3361 Epoch 82/300 Batch 0015/15, Loss: 94.1389, rec_loss: 93.7262, KL_loss: 0.0933, MMD_loss: 0.3194 Epoch 83/300 Batch 0000/15, Loss: 91.9809, rec_loss: 91.5998, KL_loss: 0.0923, MMD_loss: 0.2887 Epoch 83/300 Batch 0015/15, Loss: 92.3535, rec_loss: 91.8716, KL_loss: 0.0952, MMD_loss: 0.3867 Epoch 84/300 Batch 0000/15, Loss: 89.7356, rec_loss: 89.3484, KL_loss: 0.0945, MMD_loss: 0.2926 Epoch 84/300 Batch 0015/15, Loss: 93.0773, rec_loss: 92.7189, KL_loss: 0.0923, MMD_loss: 0.2661 Epoch 85/300 Batch 0000/15, Loss: 90.7611, rec_loss: 90.2848, KL_loss: 0.0942, MMD_loss: 0.3821 Epoch 85/300 Batch 0015/15, Loss: 90.2413, rec_loss: 89.8512, KL_loss: 0.0946, MMD_loss: 0.2954 Epoch 86/300 Batch 0000/15, Loss: 91.0277, rec_loss: 90.5886, KL_loss: 0.0929, MMD_loss: 0.3461 Epoch 86/300 Batch 0015/15, Loss: 90.9849, rec_loss: 90.5756, KL_loss: 0.0926, MMD_loss: 0.3167 Epoch 87/300 Batch 0000/15, Loss: 85.8787, rec_loss: 85.4671, KL_loss: 0.0941, MMD_loss: 0.3175 Epoch 87/300 Batch 0015/15, Loss: 89.6560, rec_loss: 89.1865, KL_loss: 0.0939, MMD_loss: 0.3756 Epoch 88/300 Batch 0000/15, Loss: 88.7800, rec_loss: 88.3320, KL_loss: 0.0980, MMD_loss: 0.3500 Epoch 88/300 Batch 0015/15, Loss: 90.0169, rec_loss: 89.6208, KL_loss: 0.0977, MMD_loss: 0.2984 Epoch 89/300 Batch 0000/15, Loss: 89.0271, rec_loss: 88.6771, KL_loss: 0.0956, MMD_loss: 0.2545 Epoch 89/300 Batch 0015/15, Loss: 87.3665, rec_loss: 86.8414, KL_loss: 0.0969, MMD_loss: 0.4282 Epoch 90/300 Batch 0000/15, Loss: 89.0134, rec_loss: 88.5822, KL_loss: 0.0939, MMD_loss: 0.3373 Epoch 90/300 Batch 0015/15, Loss: 89.4859, rec_loss: 89.1114, KL_loss: 0.0927, MMD_loss: 0.2818 Epoch 91/300 Batch 0000/15, Loss: 89.8488, rec_loss: 89.4461, KL_loss: 0.0959, MMD_loss: 0.3068 Epoch 91/300 Batch 0015/15, Loss: 90.0553, rec_loss: 89.6065, KL_loss: 0.0985, MMD_loss: 0.3502 Epoch 92/300 Batch 0000/15, Loss: 89.9612, rec_loss: 89.5611, KL_loss: 0.0985, MMD_loss: 0.3016 Epoch 92/300 Batch 0015/15, Loss: 90.3577, rec_loss: 89.9429, KL_loss: 0.1001, MMD_loss: 0.3146 Epoch 93/300 Batch 0000/15, Loss: 93.7081, rec_loss: 93.3163, KL_loss: 0.0969, MMD_loss: 0.2950 Epoch 93/300 Batch 0015/15, Loss: 89.3630, rec_loss: 88.9810, KL_loss: 0.0947, MMD_loss: 0.2873 Epoch 94/300 Batch 0000/15, Loss: 90.6467, rec_loss: 90.2616, KL_loss: 0.0991, MMD_loss: 0.2860 Epoch 94/300 Batch 0015/15, Loss: 89.7288, rec_loss: 89.3812, KL_loss: 0.0992, MMD_loss: 0.2484 Epoch 95/300 Batch 0000/15, Loss: 89.3269, rec_loss: 88.9480, KL_loss: 0.1005, MMD_loss: 0.2784 Epoch 95/300 Batch 0015/15, Loss: 88.7673, rec_loss: 88.3979, KL_loss: 0.1035, MMD_loss: 0.2659 Epoch 96/300 Batch 0000/15, Loss: 91.4750, rec_loss: 91.1085, KL_loss: 0.1013, MMD_loss: 0.2653 Epoch 96/300 Batch 0015/15, Loss: 87.8191, rec_loss: 87.4044, KL_loss: 0.1015, MMD_loss: 0.3133 Epoch 97/300 Batch 0000/15, Loss: 90.0185, rec_loss: 89.5151, KL_loss: 0.1055, MMD_loss: 0.3979 Epoch 97/300 Batch 0015/15, Loss: 94.3643, rec_loss: 93.9944, KL_loss: 0.1009, MMD_loss: 0.2690 Epoch 98/300 Batch 0000/15, Loss: 90.0230, rec_loss: 89.6199, KL_loss: 0.1029, MMD_loss: 0.3003 Epoch 98/300 Batch 0015/15, Loss: 93.6980, rec_loss: 93.3611, KL_loss: 0.1042, MMD_loss: 0.2327 Epoch 99/300 Batch 0000/15, Loss: 90.6788, rec_loss: 90.2821, KL_loss: 0.1016, MMD_loss: 0.2952 Epoch 99/300 Batch 0015/15, Loss: 86.1476, rec_loss: 85.6945, KL_loss: 0.1042, MMD_loss: 0.3490 Epoch 100/300 Batch 0000/15, Loss: 91.2125, rec_loss: 90.7201, KL_loss: 0.1058, MMD_loss: 0.3866 Epoch 100/300 Batch 0015/15, Loss: 91.0489, rec_loss: 90.6587, KL_loss: 0.1045, MMD_loss: 0.2857 Epoch 100, Loss_valid: 128.5286, rec_loss_valid: 122.1860, KL_loss_valid: 0.1415, MMD_loss: 3.7760 Epoch 101/300 Batch 0000/15, Loss: 90.5503, rec_loss: 90.2031, KL_loss: 0.1030, MMD_loss: 0.2443 Epoch 101/300 Batch 0015/15, Loss: 88.6107, rec_loss: 88.2547, KL_loss: 0.1036, MMD_loss: 0.2524 Epoch 102/300 Batch 0000/15, Loss: 92.2054, rec_loss: 91.8018, KL_loss: 0.1034, MMD_loss: 0.3002 Epoch 102/300 Batch 0015/15, Loss: 87.3683, rec_loss: 86.7653, KL_loss: 0.1018, MMD_loss: 0.5012 Epoch 103/300 Batch 0000/15, Loss: 88.3684, rec_loss: 88.0334, KL_loss: 0.1013, MMD_loss: 0.2337 Epoch 103/300 Batch 0015/15, Loss: 88.2429, rec_loss: 87.8274, KL_loss: 0.1061, MMD_loss: 0.3094 Epoch 104/300 Batch 0000/15, Loss: 88.1113, rec_loss: 87.7372, KL_loss: 0.1054, MMD_loss: 0.2687 Epoch 104/300 Batch 0015/15, Loss: 89.8001, rec_loss: 89.3401, KL_loss: 0.1080, MMD_loss: 0.3519 Epoch 105/300 Batch 0000/15, Loss: 87.8315, rec_loss: 87.4366, KL_loss: 0.1037, MMD_loss: 0.2912 Epoch 105/300 Batch 0015/15, Loss: 91.4892, rec_loss: 91.1186, KL_loss: 0.1053, MMD_loss: 0.2654 Epoch 106/300 Batch 0000/15, Loss: 88.4432, rec_loss: 88.0351, KL_loss: 0.1069, MMD_loss: 0.3012 Epoch 106/300 Batch 0015/15, Loss: 93.6184, rec_loss: 93.1537, KL_loss: 0.1050, MMD_loss: 0.3597 Epoch 107/300 Batch 0000/15, Loss: 90.2500, rec_loss: 89.8318, KL_loss: 0.1085, MMD_loss: 0.3098 Epoch 107/300 Batch 0015/15, Loss: 89.5173, rec_loss: 89.2185, KL_loss: 0.1072, MMD_loss: 0.1916 Epoch 108/300 Batch 0000/15, Loss: 89.8634, rec_loss: 89.3966, KL_loss: 0.1074, MMD_loss: 0.3595 Epoch 108/300 Batch 0015/15, Loss: 88.7810, rec_loss: 88.3306, KL_loss: 0.1071, MMD_loss: 0.3432 Epoch 109/300 Batch 0000/15, Loss: 90.3845, rec_loss: 89.9268, KL_loss: 0.1086, MMD_loss: 0.3490 Epoch 109/300 Batch 0015/15, Loss: 88.2346, rec_loss: 87.7820, KL_loss: 0.1086, MMD_loss: 0.3440 Epoch 110/300 Batch 0000/15, Loss: 87.9095, rec_loss: 87.4922, KL_loss: 0.1090, MMD_loss: 0.3083 Epoch 110/300 Batch 0015/15, Loss: 88.7859, rec_loss: 88.3493, KL_loss: 0.1106, MMD_loss: 0.3261 Epoch 111/300 Batch 0000/15, Loss: 87.6091, rec_loss: 87.1905, KL_loss: 0.1091, MMD_loss: 0.3094 Epoch 111/300 Batch 0015/15, Loss: 89.0433, rec_loss: 88.7299, KL_loss: 0.1076, MMD_loss: 0.2058 Epoch 112/300 Batch 0000/15, Loss: 87.1827, rec_loss: 86.7882, KL_loss: 0.1081, MMD_loss: 0.2864 Epoch 112/300 Batch 0015/15, Loss: 88.4161, rec_loss: 88.0596, KL_loss: 0.1090, MMD_loss: 0.2474 Epoch 113/300 Batch 0000/15, Loss: 90.8628, rec_loss: 90.4534, KL_loss: 0.1114, MMD_loss: 0.2980 Epoch 113/300 Batch 0015/15, Loss: 91.4313, rec_loss: 90.9931, KL_loss: 0.1101, MMD_loss: 0.3280 Epoch 114/300 Batch 0000/15, Loss: 89.0241, rec_loss: 88.6243, KL_loss: 0.1073, MMD_loss: 0.2925 Epoch 114/300 Batch 0015/15, Loss: 88.0171, rec_loss: 87.5845, KL_loss: 0.1097, MMD_loss: 0.3229 Epoch 115/300 Batch 0000/15, Loss: 87.7467, rec_loss: 87.4302, KL_loss: 0.1083, MMD_loss: 0.2083 Epoch 115/300 Batch 0015/15, Loss: 88.7629, rec_loss: 88.3505, KL_loss: 0.1128, MMD_loss: 0.2997 Epoch 116/300 Batch 0000/15, Loss: 90.1939, rec_loss: 89.8031, KL_loss: 0.1102, MMD_loss: 0.2806 Epoch 116/300 Batch 0015/15, Loss: 88.7055, rec_loss: 88.3660, KL_loss: 0.1125, MMD_loss: 0.2270 Epoch 117/300 Batch 0000/15, Loss: 87.8846, rec_loss: 87.5487, KL_loss: 0.1128, MMD_loss: 0.2231 Epoch 117/300 Batch 0015/15, Loss: 92.2975, rec_loss: 91.8634, KL_loss: 0.1108, MMD_loss: 0.3233 Epoch 118/300 Batch 0000/15, Loss: 89.1429, rec_loss: 88.7871, KL_loss: 0.1106, MMD_loss: 0.2452 Epoch 118/300 Batch 0015/15, Loss: 85.0030, rec_loss: 84.6192, KL_loss: 0.1108, MMD_loss: 0.2730 Epoch 119/300 Batch 0000/15, Loss: 90.6764, rec_loss: 90.3253, KL_loss: 0.1108, MMD_loss: 0.2402 Epoch 119/300 Batch 0015/15, Loss: 90.7650, rec_loss: 90.3730, KL_loss: 0.1125, MMD_loss: 0.2795 Epoch 120/300 Batch 0000/15, Loss: 87.7364, rec_loss: 87.3324, KL_loss: 0.1099, MMD_loss: 0.2941 Epoch 120/300 Batch 0015/15, Loss: 91.0855, rec_loss: 90.7539, KL_loss: 0.1104, MMD_loss: 0.2212 Epoch 120, Loss_valid: 127.5483, rec_loss_valid: 121.5923, KL_loss_valid: 0.1555, MMD_loss: 3.8274 Epoch 121/300 Batch 0000/15, Loss: 89.2908, rec_loss: 88.9319, KL_loss: 0.1119, MMD_loss: 0.2471 Epoch 121/300 Batch 0015/15, Loss: 87.2757, rec_loss: 86.9007, KL_loss: 0.1085, MMD_loss: 0.2666 Epoch 122/300 Batch 0000/15, Loss: 87.5590, rec_loss: 87.1478, KL_loss: 0.1161, MMD_loss: 0.2952 Epoch 122/300 Batch 0015/15, Loss: 89.0756, rec_loss: 88.7392, KL_loss: 0.1120, MMD_loss: 0.2244 Epoch 123/300 Batch 0000/15, Loss: 85.5059, rec_loss: 85.1318, KL_loss: 0.1116, MMD_loss: 0.2625 Epoch 123/300 Batch 0015/15, Loss: 90.9548, rec_loss: 90.6146, KL_loss: 0.1117, MMD_loss: 0.2285 Epoch 124/300 Batch 0000/15, Loss: 88.2299, rec_loss: 87.8696, KL_loss: 0.1120, MMD_loss: 0.2484 Epoch 124/300 Batch 0015/15, Loss: 88.1978, rec_loss: 87.7168, KL_loss: 0.1129, MMD_loss: 0.3681 Epoch 125/300 Batch 0000/15, Loss: 88.1732, rec_loss: 87.8145, KL_loss: 0.1132, MMD_loss: 0.2456 Epoch 125/300 Batch 0015/15, Loss: 89.2189, rec_loss: 88.9122, KL_loss: 0.1160, MMD_loss: 0.1908 Epoch 126/300 Batch 0000/15, Loss: 88.2878, rec_loss: 87.8653, KL_loss: 0.1140, MMD_loss: 0.3085 Epoch 126/300 Batch 0015/15, Loss: 88.2460, rec_loss: 87.8750, KL_loss: 0.1155, MMD_loss: 0.2556 Epoch 127/300 Batch 0000/15, Loss: 88.3410, rec_loss: 87.9899, KL_loss: 0.1125, MMD_loss: 0.2387 Epoch 127/300 Batch 0015/15, Loss: 88.7271, rec_loss: 88.3642, KL_loss: 0.1160, MMD_loss: 0.2468 Epoch 128/300 Batch 0000/15, Loss: 87.0356, rec_loss: 86.6453, KL_loss: 0.1202, MMD_loss: 0.2702 Epoch 128/300 Batch 0015/15, Loss: 86.0290, rec_loss: 85.5755, KL_loss: 0.1197, MMD_loss: 0.3337 Epoch 129/300 Batch 0000/15, Loss: 88.9025, rec_loss: 88.5112, KL_loss: 0.1165, MMD_loss: 0.2748 Epoch 129/300 Batch 0015/15, Loss: 92.2843, rec_loss: 91.9315, KL_loss: 0.1176, MMD_loss: 0.2352 Epoch 130/300 Batch 0000/15, Loss: 87.2924, rec_loss: 86.9573, KL_loss: 0.1194, MMD_loss: 0.2158 Epoch 130/300 Batch 0015/15, Loss: 90.5726, rec_loss: 90.1363, KL_loss: 0.1174, MMD_loss: 0.3188 Epoch 131/300 Batch 0000/15, Loss: 90.5163, rec_loss: 90.1232, KL_loss: 0.1179, MMD_loss: 0.2752 Epoch 131/300 Batch 0015/15, Loss: 87.7059, rec_loss: 87.4027, KL_loss: 0.1149, MMD_loss: 0.1883 Epoch 132/300 Batch 0000/15, Loss: 88.7586, rec_loss: 88.4262, KL_loss: 0.1192, MMD_loss: 0.2132 Epoch 132/300 Batch 0015/15, Loss: 89.1920, rec_loss: 88.8027, KL_loss: 0.1172, MMD_loss: 0.2722 Epoch 133/300 Batch 0000/15, Loss: 88.6928, rec_loss: 88.3459, KL_loss: 0.1207, MMD_loss: 0.2261 Epoch 133/300 Batch 0015/15, Loss: 90.9414, rec_loss: 90.5504, KL_loss: 0.1166, MMD_loss: 0.2743 Epoch 134/300 Batch 0000/15, Loss: 89.9720, rec_loss: 89.5618, KL_loss: 0.1198, MMD_loss: 0.2904 Epoch 134/300 Batch 0015/15, Loss: 90.3130, rec_loss: 89.9131, KL_loss: 0.1178, MMD_loss: 0.2821 Epoch 135/300 Batch 0000/15, Loss: 88.3656, rec_loss: 87.9372, KL_loss: 0.1206, MMD_loss: 0.3078 Epoch 135/300 Batch 0015/15, Loss: 88.9705, rec_loss: 88.6472, KL_loss: 0.1180, MMD_loss: 0.2053 Epoch 136/300 Batch 0000/15, Loss: 89.7667, rec_loss: 89.4369, KL_loss: 0.1157, MMD_loss: 0.2142 Epoch 136/300 Batch 0015/15, Loss: 88.8610, rec_loss: 88.5456, KL_loss: 0.1233, MMD_loss: 0.1921 Epoch 137/300 Batch 0000/15, Loss: 90.0844, rec_loss: 89.7258, KL_loss: 0.1185, MMD_loss: 0.2401 Epoch 137/300 Batch 0015/15, Loss: 88.9036, rec_loss: 88.6384, KL_loss: 0.1212, MMD_loss: 0.1440 Epoch 138/300 Batch 0000/15, Loss: 88.3927, rec_loss: 87.9791, KL_loss: 0.1234, MMD_loss: 0.2902 Epoch 138/300 Batch 0015/15, Loss: 89.5292, rec_loss: 89.1558, KL_loss: 0.1191, MMD_loss: 0.2544 Epoch 139/300 Batch 0000/15, Loss: 88.1120, rec_loss: 87.7328, KL_loss: 0.1200, MMD_loss: 0.2592 Epoch 139/300 Batch 0015/15, Loss: 89.5844, rec_loss: 89.2171, KL_loss: 0.1230, MMD_loss: 0.2442 Epoch 140/300 Batch 0000/15, Loss: 90.3349, rec_loss: 89.9604, KL_loss: 0.1217, MMD_loss: 0.2528 Epoch 140/300 Batch 0015/15, Loss: 91.1501, rec_loss: 90.7753, KL_loss: 0.1237, MMD_loss: 0.2510 Epoch 140, Loss_valid: 126.5580, rec_loss_valid: 120.4248, KL_loss_valid: 0.1702, MMD_loss: 3.9073 Epoch 141/300 Batch 0000/15, Loss: 89.3135, rec_loss: 88.9442, KL_loss: 0.1197, MMD_loss: 0.2496 Epoch 141/300 Batch 0015/15, Loss: 88.6364, rec_loss: 88.2278, KL_loss: 0.1175, MMD_loss: 0.2911 Epoch 142/300 Batch 0000/15, Loss: 88.1951, rec_loss: 87.8624, KL_loss: 0.1199, MMD_loss: 0.2128 Epoch 142/300 Batch 0015/15, Loss: 88.4603, rec_loss: 88.0414, KL_loss: 0.1218, MMD_loss: 0.2971 Epoch 143/300 Batch 0000/15, Loss: 89.8602, rec_loss: 89.5188, KL_loss: 0.1204, MMD_loss: 0.2210 Epoch 143/300 Batch 0015/15, Loss: 85.8937, rec_loss: 85.4749, KL_loss: 0.1203, MMD_loss: 0.2985 Epoch 144/300 Batch 0000/15, Loss: 86.5974, rec_loss: 86.1573, KL_loss: 0.1210, MMD_loss: 0.3191 Epoch 144/300 Batch 0015/15, Loss: 86.8744, rec_loss: 86.5107, KL_loss: 0.1179, MMD_loss: 0.2459 Epoch 145/300 Batch 0000/15, Loss: 89.4886, rec_loss: 89.1479, KL_loss: 0.1178, MMD_loss: 0.2229 Epoch 145/300 Batch 0015/15, Loss: 88.6853, rec_loss: 88.2929, KL_loss: 0.1213, MMD_loss: 0.2711 Epoch 146/300 Batch 0000/15, Loss: 85.0950, rec_loss: 84.6185, KL_loss: 0.1235, MMD_loss: 0.3530 Epoch 146/300 Batch 0015/15, Loss: 89.4926, rec_loss: 89.1683, KL_loss: 0.1252, MMD_loss: 0.1990 Epoch 147/300 Batch 0000/15, Loss: 89.5236, rec_loss: 89.2025, KL_loss: 0.1199, MMD_loss: 0.2012 Epoch 147/300 Batch 0015/15, Loss: 88.5093, rec_loss: 88.1444, KL_loss: 0.1213, MMD_loss: 0.2436 Epoch 148/300 Batch 0000/15, Loss: 89.1446, rec_loss: 88.7670, KL_loss: 0.1230, MMD_loss: 0.2546 Epoch 148/300 Batch 0015/15, Loss: 90.2513, rec_loss: 89.8243, KL_loss: 0.1278, MMD_loss: 0.2991 Epoch 149/300 Batch 0000/15, Loss: 86.8023, rec_loss: 86.3759, KL_loss: 0.1206, MMD_loss: 0.3058 Epoch 149/300 Batch 0015/15, Loss: 91.1062, rec_loss: 90.6061, KL_loss: 0.1224, MMD_loss: 0.3777 Epoch 150/300 Batch 0000/15, Loss: 87.4548, rec_loss: 87.0979, KL_loss: 0.1222, MMD_loss: 0.2347 Epoch 150/300 Batch 0015/15, Loss: 87.5407, rec_loss: 87.2262, KL_loss: 0.1234, MMD_loss: 0.1911 Epoch 151/300 Batch 0000/15, Loss: 89.8401, rec_loss: 89.4523, KL_loss: 0.1234, MMD_loss: 0.2645 Epoch 151/300 Batch 0015/15, Loss: 85.9707, rec_loss: 85.4893, KL_loss: 0.1227, MMD_loss: 0.3587 Epoch 152/300 Batch 0000/15, Loss: 90.4187, rec_loss: 90.0631, KL_loss: 0.1245, MMD_loss: 0.2311 Epoch 152/300 Batch 0015/15, Loss: 87.0506, rec_loss: 86.7068, KL_loss: 0.1245, MMD_loss: 0.2193 Epoch 153/300 Batch 0000/15, Loss: 87.5748, rec_loss: 87.2570, KL_loss: 0.1260, MMD_loss: 0.1918 Epoch 153/300 Batch 0015/15, Loss: 89.8107, rec_loss: 89.5165, KL_loss: 0.1220, MMD_loss: 0.1721 Epoch 154/300 Batch 0000/15, Loss: 87.6464, rec_loss: 87.2983, KL_loss: 0.1223, MMD_loss: 0.2258 Epoch 154/300 Batch 0015/15, Loss: 89.3731, rec_loss: 89.0122, KL_loss: 0.1266, MMD_loss: 0.2344 Epoch 155/300 Batch 0000/15, Loss: 87.4869, rec_loss: 87.1701, KL_loss: 0.1257, MMD_loss: 0.1912 Epoch 155/300 Batch 0015/15, Loss: 88.0939, rec_loss: 87.7582, KL_loss: 0.1285, MMD_loss: 0.2072 Epoch 156/300 Batch 0000/15, Loss: 86.1788, rec_loss: 85.8459, KL_loss: 0.1293, MMD_loss: 0.2036 Epoch 156/300 Batch 0015/15, Loss: 90.0082, rec_loss: 89.6387, KL_loss: 0.1278, MMD_loss: 0.2417 Epoch 157/300 Batch 0000/15, Loss: 88.3439, rec_loss: 88.0302, KL_loss: 0.1274, MMD_loss: 0.1863 Epoch 157/300 Batch 0015/15, Loss: 85.9087, rec_loss: 85.5724, KL_loss: 0.1268, MMD_loss: 0.2095 Epoch 158/300 Batch 0000/15, Loss: 87.7266, rec_loss: 87.3521, KL_loss: 0.1255, MMD_loss: 0.2490 Epoch 158/300 Batch 0015/15, Loss: 92.3338, rec_loss: 92.0211, KL_loss: 0.1292, MMD_loss: 0.1834 Epoch 159/300 Batch 0000/15, Loss: 89.2891, rec_loss: 88.9120, KL_loss: 0.1253, MMD_loss: 0.2518 Epoch 159/300 Batch 0015/15, Loss: 88.0859, rec_loss: 87.7340, KL_loss: 0.1309, MMD_loss: 0.2209 Epoch 160/300 Batch 0000/15, Loss: 87.1688, rec_loss: 86.7196, KL_loss: 0.1224, MMD_loss: 0.3268 Epoch 160/300 Batch 0015/15, Loss: 89.3347, rec_loss: 88.9292, KL_loss: 0.1242, MMD_loss: 0.2813 Epoch 160, Loss_valid: 126.2264, rec_loss_valid: 120.2713, KL_loss_valid: 0.1736, MMD_loss: 4.2306 Epoch 161/300 Batch 0000/15, Loss: 89.7916, rec_loss: 89.4058, KL_loss: 0.1282, MMD_loss: 0.2576 Epoch 161/300 Batch 0015/15, Loss: 90.2983, rec_loss: 89.9666, KL_loss: 0.1271, MMD_loss: 0.2046 Epoch 162/300 Batch 0000/15, Loss: 88.9470, rec_loss: 88.5775, KL_loss: 0.1268, MMD_loss: 0.2427 Epoch 162/300 Batch 0015/15, Loss: 85.2226, rec_loss: 84.9323, KL_loss: 0.1309, MMD_loss: 0.1595 Epoch 163/300 Batch 0000/15, Loss: 86.0770, rec_loss: 85.6765, KL_loss: 0.1281, MMD_loss: 0.2724 Epoch 163/300 Batch 0015/15, Loss: 88.6745, rec_loss: 88.2648, KL_loss: 0.1327, MMD_loss: 0.2770 Epoch 164/300 Batch 0000/15, Loss: 86.9365, rec_loss: 86.6084, KL_loss: 0.1308, MMD_loss: 0.1973 Epoch 164/300 Batch 0015/15, Loss: 89.9820, rec_loss: 89.5923, KL_loss: 0.1269, MMD_loss: 0.2627 Epoch 165/300 Batch 0000/15, Loss: 88.0868, rec_loss: 87.7519, KL_loss: 0.1311, MMD_loss: 0.2038 Epoch 165/300 Batch 0015/15, Loss: 90.5309, rec_loss: 90.1790, KL_loss: 0.1291, MMD_loss: 0.2229 Epoch 166/300 Batch 0000/15, Loss: 86.9595, rec_loss: 86.6016, KL_loss: 0.1246, MMD_loss: 0.2333 Epoch 166/300 Batch 0015/15, Loss: 90.8941, rec_loss: 90.4976, KL_loss: 0.1290, MMD_loss: 0.2674 Epoch 167/300 Batch 0000/15, Loss: 87.8583, rec_loss: 87.5056, KL_loss: 0.1299, MMD_loss: 0.2228 Epoch 167/300 Batch 0015/15, Loss: 85.9312, rec_loss: 85.5186, KL_loss: 0.1338, MMD_loss: 0.2789 Epoch 168/300 Batch 0000/15, Loss: 86.4583, rec_loss: 86.0924, KL_loss: 0.1261, MMD_loss: 0.2398 Epoch 168/300 Batch 0015/15, Loss: 86.6182, rec_loss: 86.2894, KL_loss: 0.1318, MMD_loss: 0.1971 Epoch 169/300 Batch 0000/15, Loss: 86.3219, rec_loss: 85.9717, KL_loss: 0.1275, MMD_loss: 0.2227 Epoch 169/300 Batch 0015/15, Loss: 88.9146, rec_loss: 88.5426, KL_loss: 0.1253, MMD_loss: 0.2467 Epoch 170/300 Batch 0000/15, Loss: 89.7609, rec_loss: 89.3936, KL_loss: 0.1312, MMD_loss: 0.2361 Epoch 170/300 Batch 0015/15, Loss: 89.3019, rec_loss: 88.9514, KL_loss: 0.1305, MMD_loss: 0.2200 Epoch 171/300 Batch 0000/15, Loss: 91.0778, rec_loss: 90.7329, KL_loss: 0.1336, MMD_loss: 0.2113 Epoch 171/300 Batch 0015/15, Loss: 86.9147, rec_loss: 86.6018, KL_loss: 0.1283, MMD_loss: 0.1846 Epoch 172/300 Batch 0000/15, Loss: 86.3956, rec_loss: 86.0424, KL_loss: 0.1284, MMD_loss: 0.2248 Epoch 172/300 Batch 0015/15, Loss: 85.7617, rec_loss: 85.4333, KL_loss: 0.1278, MMD_loss: 0.2006 Epoch 173/300 Batch 0000/15, Loss: 88.2717, rec_loss: 87.9224, KL_loss: 0.1274, MMD_loss: 0.2219 Epoch 173/300 Batch 0015/15, Loss: 90.8000, rec_loss: 90.5048, KL_loss: 0.1335, MMD_loss: 0.1618 Epoch 174/300 Batch 0000/15, Loss: 85.0268, rec_loss: 84.6709, KL_loss: 0.1297, MMD_loss: 0.2261 Epoch 174/300 Batch 0015/15, Loss: 85.1140, rec_loss: 84.7372, KL_loss: 0.1311, MMD_loss: 0.2457 Epoch 175/300 Batch 0000/15, Loss: 87.5796, rec_loss: 87.1231, KL_loss: 0.1308, MMD_loss: 0.3257 Epoch 175/300 Batch 0015/15, Loss: 86.8532, rec_loss: 86.4276, KL_loss: 0.1323, MMD_loss: 0.2934 Epoch 176/300 Batch 0000/15, Loss: 85.9746, rec_loss: 85.5435, KL_loss: 0.1317, MMD_loss: 0.2993 Epoch 176/300 Batch 0015/15, Loss: 88.6991, rec_loss: 88.3107, KL_loss: 0.1285, MMD_loss: 0.2600 Epoch 177/300 Batch 0000/15, Loss: 89.0990, rec_loss: 88.7472, KL_loss: 0.1311, MMD_loss: 0.2207 Epoch 177/300 Batch 0015/15, Loss: 86.5646, rec_loss: 86.2345, KL_loss: 0.1335, MMD_loss: 0.1966 Epoch 178/300 Batch 0000/15, Loss: 90.8889, rec_loss: 90.5713, KL_loss: 0.1360, MMD_loss: 0.1816 Epoch 178/300 Batch 0015/15, Loss: 85.7043, rec_loss: 85.2819, KL_loss: 0.1310, MMD_loss: 0.2914 Epoch 179/300 Batch 0000/15, Loss: 87.3548, rec_loss: 87.0078, KL_loss: 0.1316, MMD_loss: 0.2153 Epoch 179/300 Batch 0015/15, Loss: 84.0229, rec_loss: 83.7026, KL_loss: 0.1365, MMD_loss: 0.1837 Epoch 180/300 Batch 0000/15, Loss: 85.9664, rec_loss: 85.5246, KL_loss: 0.1347, MMD_loss: 0.3072 Epoch 180/300 Batch 0015/15, Loss: 86.9704, rec_loss: 86.5538, KL_loss: 0.1352, MMD_loss: 0.2814 Epoch 180, Loss_valid: 125.1221, rec_loss_valid: 119.5832, KL_loss_valid: 0.1834, MMD_loss: 3.7247 Epoch 181/300 Batch 0000/15, Loss: 84.9083, rec_loss: 84.4811, KL_loss: 0.1332, MMD_loss: 0.2941 Epoch 181/300 Batch 0015/15, Loss: 87.3144, rec_loss: 86.9993, KL_loss: 0.1370, MMD_loss: 0.1781 Epoch 182/300 Batch 0000/15, Loss: 87.8895, rec_loss: 87.4660, KL_loss: 0.1327, MMD_loss: 0.2908 Epoch 182/300 Batch 0015/15, Loss: 83.8019, rec_loss: 83.4654, KL_loss: 0.1350, MMD_loss: 0.2015 Epoch 183/300 Batch 0000/15, Loss: 83.3539, rec_loss: 82.9707, KL_loss: 0.1344, MMD_loss: 0.2488 Epoch 183/300 Batch 0015/15, Loss: 89.7860, rec_loss: 89.4328, KL_loss: 0.1364, MMD_loss: 0.2169 Epoch 184/300 Batch 0000/15, Loss: 86.5803, rec_loss: 86.2503, KL_loss: 0.1369, MMD_loss: 0.1931 Epoch 184/300 Batch 0015/15, Loss: 85.3249, rec_loss: 84.9505, KL_loss: 0.1332, MMD_loss: 0.2413 Epoch 185/300 Batch 0000/15, Loss: 86.7897, rec_loss: 86.4212, KL_loss: 0.1346, MMD_loss: 0.2339 Epoch 185/300 Batch 0015/15, Loss: 85.7548, rec_loss: 85.3941, KL_loss: 0.1356, MMD_loss: 0.2250 Epoch 186/300 Batch 0000/15, Loss: 86.2554, rec_loss: 85.8820, KL_loss: 0.1340, MMD_loss: 0.2394 Epoch 186/300 Batch 0015/15, Loss: 85.6995, rec_loss: 85.3450, KL_loss: 0.1297, MMD_loss: 0.2248 Epoch 187/300 Batch 0000/15, Loss: 88.7440, rec_loss: 88.4054, KL_loss: 0.1328, MMD_loss: 0.2058 Epoch 187/300 Batch 0015/15, Loss: 86.0549, rec_loss: 85.7530, KL_loss: 0.1335, MMD_loss: 0.1685 Epoch 188/300 Batch 0000/15, Loss: 85.2332, rec_loss: 84.9189, KL_loss: 0.1355, MMD_loss: 0.1789 Epoch 188/300 Batch 0015/15, Loss: 89.6851, rec_loss: 89.2290, KL_loss: 0.1325, MMD_loss: 0.3235 Epoch 189/300 Batch 0000/15, Loss: 85.5008, rec_loss: 85.1507, KL_loss: 0.1352, MMD_loss: 0.2149 Epoch 189/300 Batch 0015/15, Loss: 90.4378, rec_loss: 89.9906, KL_loss: 0.1322, MMD_loss: 0.3150 Epoch 190/300 Batch 0000/15, Loss: 88.6927, rec_loss: 88.3223, KL_loss: 0.1374, MMD_loss: 0.2330 Epoch 190/300 Batch 0015/15, Loss: 89.3348, rec_loss: 88.9564, KL_loss: 0.1345, MMD_loss: 0.2439 Epoch 191/300 Batch 0000/15, Loss: 86.3476, rec_loss: 86.0039, KL_loss: 0.1351, MMD_loss: 0.2086 Epoch 191/300 Batch 0015/15, Loss: 85.3554, rec_loss: 84.8988, KL_loss: 0.1370, MMD_loss: 0.3197 Epoch 192/300 Batch 0000/15, Loss: 86.7130, rec_loss: 86.3303, KL_loss: 0.1371, MMD_loss: 0.2455 Epoch 192/300 Batch 0015/15, Loss: 85.3010, rec_loss: 84.9670, KL_loss: 0.1372, MMD_loss: 0.1969 Epoch 193/300 Batch 0000/15, Loss: 84.8269, rec_loss: 84.4691, KL_loss: 0.1346, MMD_loss: 0.2231 Epoch 193/300 Batch 0015/15, Loss: 90.7365, rec_loss: 90.4243, KL_loss: 0.1354, MMD_loss: 0.1769 Epoch 194/300 Batch 0000/15, Loss: 84.5253, rec_loss: 84.1461, KL_loss: 0.1356, MMD_loss: 0.2436 Epoch 194/300 Batch 0015/15, Loss: 87.9853, rec_loss: 87.6732, KL_loss: 0.1416, MMD_loss: 0.1704 Epoch 195/300 Batch 0000/15, Loss: 84.0531, rec_loss: 83.7186, KL_loss: 0.1384, MMD_loss: 0.1961 Epoch 195/300 Batch 0015/15, Loss: 85.1213, rec_loss: 84.7920, KL_loss: 0.1342, MMD_loss: 0.1951 Epoch 196/300 Batch 0000/15, Loss: 87.5889, rec_loss: 87.2175, KL_loss: 0.1340, MMD_loss: 0.2374 Epoch 196/300 Batch 0015/15, Loss: 83.2978, rec_loss: 82.9444, KL_loss: 0.1393, MMD_loss: 0.2140 Epoch 197/300 Batch 0000/15, Loss: 85.8766, rec_loss: 85.5166, KL_loss: 0.1327, MMD_loss: 0.2272 Epoch 197/300 Batch 0015/15, Loss: 88.0670, rec_loss: 87.6445, KL_loss: 0.1362, MMD_loss: 0.2863 Epoch 198/300 Batch 0000/15, Loss: 86.2430, rec_loss: 85.8110, KL_loss: 0.1354, MMD_loss: 0.2965 Epoch 198/300 Batch 0015/15, Loss: 83.8788, rec_loss: 83.5571, KL_loss: 0.1380, MMD_loss: 0.1837 Epoch 199/300 Batch 0000/15, Loss: 84.0330, rec_loss: 83.6741, KL_loss: 0.1425, MMD_loss: 0.2164 Epoch 199/300 Batch 0015/15, Loss: 86.2519, rec_loss: 85.8789, KL_loss: 0.1428, MMD_loss: 0.2302 Epoch 200/300 Batch 0000/15, Loss: 84.0533, rec_loss: 83.7330, KL_loss: 0.1377, MMD_loss: 0.1827 Epoch 200/300 Batch 0015/15, Loss: 85.5300, rec_loss: 85.1632, KL_loss: 0.1409, MMD_loss: 0.2259 Epoch 200, Loss_valid: 124.2800, rec_loss_valid: 119.0530, KL_loss_valid: 0.1858, MMD_loss: 3.3483 Epoch 201/300 Batch 0000/15, Loss: 86.2822, rec_loss: 85.9560, KL_loss: 0.1400, MMD_loss: 0.1862 Epoch 201/300 Batch 0015/15, Loss: 89.9936, rec_loss: 89.6210, KL_loss: 0.1309, MMD_loss: 0.2417 Epoch 202/300 Batch 0000/15, Loss: 86.8724, rec_loss: 86.5253, KL_loss: 0.1359, MMD_loss: 0.2112 Epoch 202/300 Batch 0015/15, Loss: 82.9704, rec_loss: 82.6361, KL_loss: 0.1375, MMD_loss: 0.1968 Epoch 203/300 Batch 0000/15, Loss: 87.2075, rec_loss: 86.8328, KL_loss: 0.1364, MMD_loss: 0.2383 Epoch 203/300 Batch 0015/15, Loss: 88.3064, rec_loss: 87.9104, KL_loss: 0.1424, MMD_loss: 0.2536 Epoch 204/300 Batch 0000/15, Loss: 87.0690, rec_loss: 86.7269, KL_loss: 0.1418, MMD_loss: 0.2004 Epoch 204/300 Batch 0015/15, Loss: 85.9447, rec_loss: 85.6079, KL_loss: 0.1392, MMD_loss: 0.1976 Epoch 205/300 Batch 0000/15, Loss: 86.5376, rec_loss: 86.1882, KL_loss: 0.1431, MMD_loss: 0.2064 Epoch 205/300 Batch 0015/15, Loss: 88.6525, rec_loss: 88.3306, KL_loss: 0.1415, MMD_loss: 0.1805 Epoch 206/300 Batch 0000/15, Loss: 85.6330, rec_loss: 85.2722, KL_loss: 0.1434, MMD_loss: 0.2174 Epoch 206/300 Batch 0015/15, Loss: 86.4835, rec_loss: 86.1232, KL_loss: 0.1378, MMD_loss: 0.2225 Epoch 207/300 Batch 0000/15, Loss: 84.4147, rec_loss: 84.0469, KL_loss: 0.1409, MMD_loss: 0.2269 Epoch 207/300 Batch 0015/15, Loss: 85.0716, rec_loss: 84.6940, KL_loss: 0.1381, MMD_loss: 0.2395 Epoch 208/300 Batch 0000/15, Loss: 88.0116, rec_loss: 87.6433, KL_loss: 0.1416, MMD_loss: 0.2267 Epoch 208/300 Batch 0015/15, Loss: 93.5977, rec_loss: 93.1857, KL_loss: 0.1383, MMD_loss: 0.2738 Epoch 209/300 Batch 0000/15, Loss: 90.0350, rec_loss: 89.6710, KL_loss: 0.1414, MMD_loss: 0.2226 Epoch 209/300 Batch 0015/15, Loss: 86.3961, rec_loss: 86.0564, KL_loss: 0.1383, MMD_loss: 0.2013 Epoch 210/300 Batch 0000/15, Loss: 87.4505, rec_loss: 87.0760, KL_loss: 0.1406, MMD_loss: 0.2340 Epoch 210/300 Batch 0015/15, Loss: 84.5745, rec_loss: 84.2279, KL_loss: 0.1368, MMD_loss: 0.2098 Epoch 211/300 Batch 0000/15, Loss: 88.1957, rec_loss: 87.8372, KL_loss: 0.1379, MMD_loss: 0.2205 Epoch 211/300 Batch 0015/15, Loss: 84.5081, rec_loss: 84.0275, KL_loss: 0.1415, MMD_loss: 0.3391 Epoch 212/300 Batch 0000/15, Loss: 85.2223, rec_loss: 84.8610, KL_loss: 0.1427, MMD_loss: 0.2186 Epoch 212/300 Batch 0015/15, Loss: 87.0632, rec_loss: 86.6579, KL_loss: 0.1386, MMD_loss: 0.2667 Epoch 213/300 Batch 0000/15, Loss: 88.1333, rec_loss: 87.8203, KL_loss: 0.1378, MMD_loss: 0.1752 Epoch 213/300 Batch 0015/15, Loss: 85.9710, rec_loss: 85.5446, KL_loss: 0.1493, MMD_loss: 0.2771 Epoch 214/300 Batch 0000/15, Loss: 85.3925, rec_loss: 85.0419, KL_loss: 0.1420, MMD_loss: 0.2085 Epoch 214/300 Batch 0015/15, Loss: 84.2050, rec_loss: 83.8637, KL_loss: 0.1375, MMD_loss: 0.2038 Epoch 215/300 Batch 0000/15, Loss: 86.9918, rec_loss: 86.6890, KL_loss: 0.1395, MMD_loss: 0.1633 Epoch 215/300 Batch 0015/15, Loss: 87.2533, rec_loss: 86.9370, KL_loss: 0.1379, MMD_loss: 0.1783 Epoch 216/300 Batch 0000/15, Loss: 88.0970, rec_loss: 87.7339, KL_loss: 0.1410, MMD_loss: 0.2221 Epoch 216/300 Batch 0015/15, Loss: 86.9183, rec_loss: 86.5214, KL_loss: 0.1364, MMD_loss: 0.2605 Epoch 217/300 Batch 0000/15, Loss: 86.2235, rec_loss: 85.8666, KL_loss: 0.1410, MMD_loss: 0.2160 Epoch 217/300 Batch 0015/15, Loss: 84.9105, rec_loss: 84.5366, KL_loss: 0.1426, MMD_loss: 0.2313 Epoch 218/300 Batch 0000/15, Loss: 87.5593, rec_loss: 87.1741, KL_loss: 0.1417, MMD_loss: 0.2435 Epoch 218/300 Batch 0015/15, Loss: 85.5720, rec_loss: 85.2095, KL_loss: 0.1379, MMD_loss: 0.2246 Epoch 219/300 Batch 0000/15, Loss: 90.5775, rec_loss: 90.2292, KL_loss: 0.1422, MMD_loss: 0.2062 Epoch 219/300 Batch 0015/15, Loss: 85.5877, rec_loss: 85.2069, KL_loss: 0.1430, MMD_loss: 0.2378 Epoch 220/300 Batch 0000/15, Loss: 83.5327, rec_loss: 83.1530, KL_loss: 0.1443, MMD_loss: 0.2354 Epoch 220/300 Batch 0015/15, Loss: 87.7274, rec_loss: 87.3978, KL_loss: 0.1433, MMD_loss: 0.1863 Epoch 220, Loss_valid: 123.5621, rec_loss_valid: 118.7862, KL_loss_valid: 0.1914, MMD_loss: 2.7717 Epoch 221/300 Batch 0000/15, Loss: 85.3137, rec_loss: 84.9773, KL_loss: 0.1417, MMD_loss: 0.1946 Epoch 221/300 Batch 0015/15, Loss: 84.7640, rec_loss: 84.3829, KL_loss: 0.1412, MMD_loss: 0.2398 Epoch 222/300 Batch 0000/15, Loss: 85.0462, rec_loss: 84.6840, KL_loss: 0.1422, MMD_loss: 0.2200 Epoch 222/300 Batch 0015/15, Loss: 84.3695, rec_loss: 84.0000, KL_loss: 0.1419, MMD_loss: 0.2276 Epoch 223/300 Batch 0000/15, Loss: 84.6023, rec_loss: 84.2526, KL_loss: 0.1409, MMD_loss: 0.2088 Epoch 223/300 Batch 0015/15, Loss: 84.4229, rec_loss: 84.0997, KL_loss: 0.1418, MMD_loss: 0.1814 Epoch 224/300 Batch 0000/15, Loss: 85.6521, rec_loss: 85.2123, KL_loss: 0.1387, MMD_loss: 0.3012 Epoch 224/300 Batch 0015/15, Loss: 87.2424, rec_loss: 86.8129, KL_loss: 0.1487, MMD_loss: 0.2809 Epoch 225/300 Batch 0000/15, Loss: 86.8677, rec_loss: 86.4863, KL_loss: 0.1455, MMD_loss: 0.2358 Epoch 225/300 Batch 0015/15, Loss: 88.0765, rec_loss: 87.6172, KL_loss: 0.1389, MMD_loss: 0.3204 Epoch 226/300 Batch 0000/15, Loss: 83.3256, rec_loss: 82.9059, KL_loss: 0.1404, MMD_loss: 0.2793 Epoch 226/300 Batch 0015/15, Loss: 86.7891, rec_loss: 86.3846, KL_loss: 0.1382, MMD_loss: 0.2664 Epoch 227/300 Batch 0000/15, Loss: 84.5058, rec_loss: 84.1623, KL_loss: 0.1402, MMD_loss: 0.2034 Epoch 227/300 Batch 0015/15, Loss: 83.5865, rec_loss: 83.1969, KL_loss: 0.1433, MMD_loss: 0.2464 Epoch 228/300 Batch 0000/15, Loss: 83.8168, rec_loss: 83.4347, KL_loss: 0.1399, MMD_loss: 0.2423 Epoch 228/300 Batch 0015/15, Loss: 83.0476, rec_loss: 82.7061, KL_loss: 0.1362, MMD_loss: 0.2053 Epoch 229/300 Batch 0000/15, Loss: 87.7490, rec_loss: 87.3317, KL_loss: 0.1431, MMD_loss: 0.2742 Epoch 229/300 Batch 0015/15, Loss: 85.4249, rec_loss: 85.0397, KL_loss: 0.1452, MMD_loss: 0.2400 Epoch 230/300 Batch 0000/15, Loss: 86.8262, rec_loss: 86.4662, KL_loss: 0.1423, MMD_loss: 0.2178 Epoch 230/300 Batch 0015/15, Loss: 86.6766, rec_loss: 86.2835, KL_loss: 0.1418, MMD_loss: 0.2513 Epoch 231/300 Batch 0000/15, Loss: 86.7102, rec_loss: 86.4159, KL_loss: 0.1443, MMD_loss: 0.1500 Epoch 231/300 Batch 0015/15, Loss: 84.2320, rec_loss: 83.8206, KL_loss: 0.1384, MMD_loss: 0.2729 Epoch 232/300 Batch 0000/15, Loss: 88.1110, rec_loss: 87.7621, KL_loss: 0.1484, MMD_loss: 0.2005 Epoch 232/300 Batch 0015/15, Loss: 87.1834, rec_loss: 86.7691, KL_loss: 0.1448, MMD_loss: 0.2695 Epoch 233/300 Batch 0000/15, Loss: 86.5608, rec_loss: 86.2386, KL_loss: 0.1434, MMD_loss: 0.1789 Epoch 233/300 Batch 0015/15, Loss: 83.1578, rec_loss: 82.7757, KL_loss: 0.1432, MMD_loss: 0.2390 Epoch 234/300 Batch 0000/15, Loss: 90.9531, rec_loss: 90.6324, KL_loss: 0.1475, MMD_loss: 0.1732 Epoch 234/300 Batch 0015/15, Loss: 82.6081, rec_loss: 82.2609, KL_loss: 0.1467, MMD_loss: 0.2005 Epoch 235/300 Batch 0000/15, Loss: 87.6318, rec_loss: 87.3156, KL_loss: 0.1436, MMD_loss: 0.1727 Epoch 235/300 Batch 0015/15, Loss: 87.3779, rec_loss: 87.0481, KL_loss: 0.1459, MMD_loss: 0.1840 Epoch 236/300 Batch 0000/15, Loss: 86.9199, rec_loss: 86.5550, KL_loss: 0.1432, MMD_loss: 0.2218 Epoch 236/300 Batch 0015/15, Loss: 88.0344, rec_loss: 87.6946, KL_loss: 0.1444, MMD_loss: 0.1955 Epoch 237/300 Batch 0000/15, Loss: 87.6334, rec_loss: 87.3267, KL_loss: 0.1432, MMD_loss: 0.1635 Epoch 237/300 Batch 0015/15, Loss: 84.9246, rec_loss: 84.5954, KL_loss: 0.1400, MMD_loss: 0.1892 Epoch 238/300 Batch 0000/15, Loss: 84.5333, rec_loss: 84.2101, KL_loss: 0.1495, MMD_loss: 0.1737 Epoch 238/300 Batch 0015/15, Loss: 84.3543, rec_loss: 83.9933, KL_loss: 0.1438, MMD_loss: 0.2173 Epoch 239/300 Batch 0000/15, Loss: 84.1562, rec_loss: 83.7913, KL_loss: 0.1515, MMD_loss: 0.2135 Epoch 239/300 Batch 0015/15, Loss: 87.0897, rec_loss: 86.7602, KL_loss: 0.1504, MMD_loss: 0.1791 Epoch 240/300 Batch 0000/15, Loss: 85.8030, rec_loss: 85.4651, KL_loss: 0.1418, MMD_loss: 0.1961 Epoch 240/300 Batch 0015/15, Loss: 82.7740, rec_loss: 82.3948, KL_loss: 0.1512, MMD_loss: 0.2281 Epoch 240, Loss_valid: 123.8219, rec_loss_valid: 118.4634, KL_loss_valid: 0.1995, MMD_loss: 3.1537 Epoch 241/300 Batch 0000/15, Loss: 86.0661, rec_loss: 85.7260, KL_loss: 0.1472, MMD_loss: 0.1929 Epoch 241/300 Batch 0015/15, Loss: 84.4234, rec_loss: 84.0797, KL_loss: 0.1432, MMD_loss: 0.2005 Epoch 242/300 Batch 0000/15, Loss: 84.8255, rec_loss: 84.4818, KL_loss: 0.1455, MMD_loss: 0.1982 Epoch 242/300 Batch 0015/15, Loss: 85.5318, rec_loss: 85.1285, KL_loss: 0.1398, MMD_loss: 0.2635 Epoch 243/300 Batch 0000/15, Loss: 88.3046, rec_loss: 87.9592, KL_loss: 0.1483, MMD_loss: 0.1971 Epoch 243/300 Batch 0015/15, Loss: 88.5712, rec_loss: 88.1337, KL_loss: 0.1413, MMD_loss: 0.2962 Epoch 244/300 Batch 0000/15, Loss: 87.7348, rec_loss: 87.3761, KL_loss: 0.1466, MMD_loss: 0.2121 Epoch 244/300 Batch 0015/15, Loss: 87.7916, rec_loss: 87.4096, KL_loss: 0.1470, MMD_loss: 0.2351 Epoch 245/300 Batch 0000/15, Loss: 82.9217, rec_loss: 82.6270, KL_loss: 0.1486, MMD_loss: 0.1461 Epoch 245/300 Batch 0015/15, Loss: 83.2609, rec_loss: 82.9164, KL_loss: 0.1488, MMD_loss: 0.1957 Epoch 246/300 Batch 0000/15, Loss: 88.7493, rec_loss: 88.4182, KL_loss: 0.1415, MMD_loss: 0.1896 Epoch 246/300 Batch 0015/15, Loss: 82.2598, rec_loss: 81.8970, KL_loss: 0.1472, MMD_loss: 0.2157 Epoch 247/300 Batch 0000/15, Loss: 90.4163, rec_loss: 90.0831, KL_loss: 0.1395, MMD_loss: 0.1936 Epoch 247/300 Batch 0015/15, Loss: 86.3580, rec_loss: 86.0281, KL_loss: 0.1501, MMD_loss: 0.1798 Epoch 248/300 Batch 0000/15, Loss: 86.7393, rec_loss: 86.3516, KL_loss: 0.1457, MMD_loss: 0.2420 Epoch 248/300 Batch 0015/15, Loss: 87.6098, rec_loss: 87.2754, KL_loss: 0.1536, MMD_loss: 0.1808 Epoch 249/300 Batch 0000/15, Loss: 86.7256, rec_loss: 86.4075, KL_loss: 0.1482, MMD_loss: 0.1700 Epoch 249/300 Batch 0015/15, Loss: 87.0603, rec_loss: 86.7443, KL_loss: 0.1531, MMD_loss: 0.1630 Epoch 250/300 Batch 0000/15, Loss: 83.5984, rec_loss: 83.2197, KL_loss: 0.1453, MMD_loss: 0.2333 Epoch 250/300 Batch 0015/15, Loss: 84.0626, rec_loss: 83.7110, KL_loss: 0.1446, MMD_loss: 0.2070 Epoch 251/300 Batch 0000/15, Loss: 88.5731, rec_loss: 88.2176, KL_loss: 0.1491, MMD_loss: 0.2063 Epoch 251/300 Batch 0015/15, Loss: 87.9734, rec_loss: 87.6139, KL_loss: 0.1452, MMD_loss: 0.2143 Epoch 252/300 Batch 0000/15, Loss: 85.9991, rec_loss: 85.6478, KL_loss: 0.1474, MMD_loss: 0.2040 Epoch 252/300 Batch 0015/15, Loss: 84.4947, rec_loss: 84.1539, KL_loss: 0.1452, MMD_loss: 0.1956 Epoch 253/300 Batch 0000/15, Loss: 88.3712, rec_loss: 88.0013, KL_loss: 0.1476, MMD_loss: 0.2223 Epoch 253/300 Batch 0015/15, Loss: 83.6163, rec_loss: 83.3138, KL_loss: 0.1451, MMD_loss: 0.1574 Epoch 254/300 Batch 0000/15, Loss: 86.9609, rec_loss: 86.6329, KL_loss: 0.1477, MMD_loss: 0.1803 Epoch 254/300 Batch 0015/15, Loss: 87.5038, rec_loss: 87.1372, KL_loss: 0.1486, MMD_loss: 0.2180 Epoch 255/300 Batch 0000/15, Loss: 86.0919, rec_loss: 85.7701, KL_loss: 0.1447, MMD_loss: 0.1771 Epoch 255/300 Batch 0015/15, Loss: 86.8041, rec_loss: 86.4161, KL_loss: 0.1496, MMD_loss: 0.2384 Epoch 256/300 Batch 0000/15, Loss: 88.6750, rec_loss: 88.3555, KL_loss: 0.1471, MMD_loss: 0.1724 Epoch 256/300 Batch 0015/15, Loss: 89.2324, rec_loss: 88.8875, KL_loss: 0.1485, MMD_loss: 0.1964 Epoch 257/300 Batch 0000/15, Loss: 89.4353, rec_loss: 89.1117, KL_loss: 0.1437, MMD_loss: 0.1799 Epoch 257/300 Batch 0015/15, Loss: 87.6113, rec_loss: 87.2150, KL_loss: 0.1478, MMD_loss: 0.2485 Epoch 258/300 Batch 0000/15, Loss: 85.6923, rec_loss: 85.3545, KL_loss: 0.1503, MMD_loss: 0.1875 Epoch 258/300 Batch 0015/15, Loss: 85.1296, rec_loss: 84.7987, KL_loss: 0.1505, MMD_loss: 0.1804 Epoch 259/300 Batch 0000/15, Loss: 84.8438, rec_loss: 84.5101, KL_loss: 0.1508, MMD_loss: 0.1829 Epoch 259/300 Batch 0015/15, Loss: 85.5994, rec_loss: 85.2819, KL_loss: 0.1481, MMD_loss: 0.1694 Epoch 260/300 Batch 0000/15, Loss: 81.1966, rec_loss: 80.8510, KL_loss: 0.1514, MMD_loss: 0.1942 Epoch 260/300 Batch 0015/15, Loss: 87.2830, rec_loss: 86.8694, KL_loss: 0.1462, MMD_loss: 0.2674 Epoch 260, Loss_valid: 123.3512, rec_loss_valid: 118.2663, KL_loss_valid: 0.1979, MMD_loss: 3.0962 Epoch 261/300 Batch 0000/15, Loss: 85.3728, rec_loss: 85.0588, KL_loss: 0.1462, MMD_loss: 0.1678 Epoch 261/300 Batch 0015/15, Loss: 87.3317, rec_loss: 87.0103, KL_loss: 0.1449, MMD_loss: 0.1766 Epoch 262/300 Batch 0000/15, Loss: 87.9121, rec_loss: 87.5727, KL_loss: 0.1490, MMD_loss: 0.1904 Epoch 262/300 Batch 0015/15, Loss: 86.2158, rec_loss: 85.7577, KL_loss: 0.1480, MMD_loss: 0.3101 Epoch 263/300 Batch 0000/15, Loss: 85.1926, rec_loss: 84.8180, KL_loss: 0.1458, MMD_loss: 0.2288 Epoch 263/300 Batch 0015/15, Loss: 87.7072, rec_loss: 87.3678, KL_loss: 0.1395, MMD_loss: 0.1999 Epoch 264/300 Batch 0000/15, Loss: 86.3010, rec_loss: 85.9485, KL_loss: 0.1460, MMD_loss: 0.2065 Epoch 264/300 Batch 0015/15, Loss: 86.5187, rec_loss: 86.0986, KL_loss: 0.1428, MMD_loss: 0.2773 Epoch 265/300 Batch 0000/15, Loss: 88.7647, rec_loss: 88.4469, KL_loss: 0.1491, MMD_loss: 0.1687 Epoch 265/300 Batch 0015/15, Loss: 86.7113, rec_loss: 86.2808, KL_loss: 0.1495, MMD_loss: 0.2811 Epoch 266/300 Batch 0000/15, Loss: 88.0041, rec_loss: 87.6655, KL_loss: 0.1509, MMD_loss: 0.1876 Epoch 266/300 Batch 0015/15, Loss: 84.2025, rec_loss: 83.7985, KL_loss: 0.1487, MMD_loss: 0.2553 Epoch 267/300 Batch 0000/15, Loss: 86.6672, rec_loss: 86.3710, KL_loss: 0.1489, MMD_loss: 0.1473 Epoch 267/300 Batch 0015/15, Loss: 86.5445, rec_loss: 86.1646, KL_loss: 0.1509, MMD_loss: 0.2290 Epoch 268/300 Batch 0000/15, Loss: 84.7915, rec_loss: 84.4355, KL_loss: 0.1441, MMD_loss: 0.2119 Epoch 268/300 Batch 0015/15, Loss: 85.5136, rec_loss: 85.1935, KL_loss: 0.1424, MMD_loss: 0.1777 Epoch 269/300 Batch 0000/15, Loss: 85.1773, rec_loss: 84.8074, KL_loss: 0.1456, MMD_loss: 0.2242 Epoch 269/300 Batch 0015/15, Loss: 88.0245, rec_loss: 87.6579, KL_loss: 0.1495, MMD_loss: 0.2171 Epoch 270/300 Batch 0000/15, Loss: 87.5601, rec_loss: 87.2053, KL_loss: 0.1451, MMD_loss: 0.2097 Epoch 270/300 Batch 0015/15, Loss: 90.6261, rec_loss: 90.2355, KL_loss: 0.1452, MMD_loss: 0.2453 Epoch 271/300 Batch 0000/15, Loss: 84.9907, rec_loss: 84.6994, KL_loss: 0.1506, MMD_loss: 0.1407 Epoch 271/300 Batch 0015/15, Loss: 87.7479, rec_loss: 87.4050, KL_loss: 0.1518, MMD_loss: 0.1911 Epoch 272/300 Batch 0000/15, Loss: 87.4030, rec_loss: 87.0404, KL_loss: 0.1520, MMD_loss: 0.2106 Epoch 272/300 Batch 0015/15, Loss: 85.9622, rec_loss: 85.6870, KL_loss: 0.1486, MMD_loss: 0.1266 Epoch 273/300 Batch 0000/15, Loss: 85.8217, rec_loss: 85.4571, KL_loss: 0.1491, MMD_loss: 0.2155 Epoch 273/300 Batch 0015/15, Loss: 86.6622, rec_loss: 86.3257, KL_loss: 0.1468, MMD_loss: 0.1898 Epoch 274/300 Batch 0000/15, Loss: 83.7913, rec_loss: 83.3705, KL_loss: 0.1486, MMD_loss: 0.2722 Epoch 274/300 Batch 0015/15, Loss: 84.3117, rec_loss: 83.9763, KL_loss: 0.1563, MMD_loss: 0.1792 Epoch 275/300 Batch 0000/15, Loss: 84.1764, rec_loss: 83.8558, KL_loss: 0.1483, MMD_loss: 0.1723 Epoch 275/300 Batch 0015/15, Loss: 87.2294, rec_loss: 86.8381, KL_loss: 0.1509, MMD_loss: 0.2404 Epoch 276/300 Batch 0000/15, Loss: 85.6623, rec_loss: 85.3322, KL_loss: 0.1515, MMD_loss: 0.1786 Epoch 276/300 Batch 0015/15, Loss: 88.3208, rec_loss: 88.0046, KL_loss: 0.1508, MMD_loss: 0.1654 Epoch 277/300 Batch 0000/15, Loss: 86.8883, rec_loss: 86.5071, KL_loss: 0.1480, MMD_loss: 0.2332 Epoch 277/300 Batch 0015/15, Loss: 87.3065, rec_loss: 86.9926, KL_loss: 0.1499, MMD_loss: 0.1641 Epoch 278/300 Batch 0000/15, Loss: 87.4449, rec_loss: 87.1045, KL_loss: 0.1516, MMD_loss: 0.1888 Epoch 278/300 Batch 0015/15, Loss: 85.6328, rec_loss: 85.2216, KL_loss: 0.1530, MMD_loss: 0.2582 Epoch 279/300 Batch 0000/15, Loss: 82.1571, rec_loss: 81.8217, KL_loss: 0.1495, MMD_loss: 0.1859 Epoch 279/300 Batch 0015/15, Loss: 87.8493, rec_loss: 87.5590, KL_loss: 0.1506, MMD_loss: 0.1396 Epoch 280/300 Batch 0000/15, Loss: 84.5547, rec_loss: 84.2167, KL_loss: 0.1479, MMD_loss: 0.1901 Epoch 280/300 Batch 0015/15, Loss: 85.9424, rec_loss: 85.6495, KL_loss: 0.1526, MMD_loss: 0.1404 Epoch 280, Loss_valid: 123.2934, rec_loss_valid: 118.1615, KL_loss_valid: 0.1997, MMD_loss: 4.2777 Epoch 281/300 Batch 0000/15, Loss: 83.8814, rec_loss: 83.5462, KL_loss: 0.1484, MMD_loss: 0.1868 Epoch 281/300 Batch 0015/15, Loss: 83.4888, rec_loss: 83.1602, KL_loss: 0.1521, MMD_loss: 0.1765 Epoch 282/300 Batch 0000/15, Loss: 87.3811, rec_loss: 87.0271, KL_loss: 0.1534, MMD_loss: 0.2006 Epoch 282/300 Batch 0015/15, Loss: 85.2023, rec_loss: 84.9016, KL_loss: 0.1537, MMD_loss: 0.1470 Epoch 283/300 Batch 0000/15, Loss: 83.9458, rec_loss: 83.6295, KL_loss: 0.1497, MMD_loss: 0.1666 Epoch 283/300 Batch 0015/15, Loss: 84.5908, rec_loss: 84.2545, KL_loss: 0.1507, MMD_loss: 0.1857 Epoch 284/300 Batch 0000/15, Loss: 84.3243, rec_loss: 83.9671, KL_loss: 0.1562, MMD_loss: 0.2010 Epoch 284/300 Batch 0015/15, Loss: 87.9117, rec_loss: 87.5375, KL_loss: 0.1483, MMD_loss: 0.2259 Epoch 285/300 Batch 0000/15, Loss: 86.9900, rec_loss: 86.6408, KL_loss: 0.1476, MMD_loss: 0.2016 Epoch 285/300 Batch 0015/15, Loss: 85.7961, rec_loss: 85.4789, KL_loss: 0.1484, MMD_loss: 0.1688 Epoch 286/300 Batch 0000/15, Loss: 82.4677, rec_loss: 82.1246, KL_loss: 0.1507, MMD_loss: 0.1925 Epoch 286/300 Batch 0015/15, Loss: 83.1407, rec_loss: 82.7665, KL_loss: 0.1520, MMD_loss: 0.2222 Epoch 287/300 Batch 0000/15, Loss: 86.4114, rec_loss: 86.0393, KL_loss: 0.1519, MMD_loss: 0.2201 Epoch 287/300 Batch 0015/15, Loss: 89.6433, rec_loss: 89.3083, KL_loss: 0.1504, MMD_loss: 0.1846 Epoch 288/300 Batch 0000/15, Loss: 82.7523, rec_loss: 82.4379, KL_loss: 0.1561, MMD_loss: 0.1584 Epoch 288/300 Batch 0015/15, Loss: 86.7235, rec_loss: 86.3410, KL_loss: 0.1510, MMD_loss: 0.2315 Epoch 289/300 Batch 0000/15, Loss: 86.2569, rec_loss: 85.9211, KL_loss: 0.1487, MMD_loss: 0.1871 Epoch 289/300 Batch 0015/15, Loss: 85.0506, rec_loss: 84.6711, KL_loss: 0.1545, MMD_loss: 0.2249 Epoch 290/300 Batch 0000/15, Loss: 82.9145, rec_loss: 82.5509, KL_loss: 0.1538, MMD_loss: 0.2097 Epoch 290/300 Batch 0015/15, Loss: 88.0485, rec_loss: 87.6932, KL_loss: 0.1497, MMD_loss: 0.2056 Epoch 291/300 Batch 0000/15, Loss: 85.7132, rec_loss: 85.3576, KL_loss: 0.1485, MMD_loss: 0.2072 Epoch 291/300 Batch 0015/15, Loss: 86.5469, rec_loss: 86.1283, KL_loss: 0.1488, MMD_loss: 0.2698 Epoch 292/300 Batch 0000/15, Loss: 83.9133, rec_loss: 83.5765, KL_loss: 0.1554, MMD_loss: 0.1815 Epoch 292/300 Batch 0015/15, Loss: 84.2012, rec_loss: 83.8158, KL_loss: 0.1472, MMD_loss: 0.2382 Epoch 293/300 Batch 0000/15, Loss: 84.2061, rec_loss: 83.8754, KL_loss: 0.1523, MMD_loss: 0.1785 Epoch 293/300 Batch 0015/15, Loss: 86.4187, rec_loss: 86.1049, KL_loss: 0.1524, MMD_loss: 0.1614 Epoch 294/300 Batch 0000/15, Loss: 86.9202, rec_loss: 86.5638, KL_loss: 0.1484, MMD_loss: 0.2080 Epoch 294/300 Batch 0015/15, Loss: 88.4378, rec_loss: 88.0569, KL_loss: 0.1515, MMD_loss: 0.2294 Epoch 295/300 Batch 0000/15, Loss: 88.3305, rec_loss: 87.9960, KL_loss: 0.1503, MMD_loss: 0.1842 Epoch 295/300 Batch 0015/15, Loss: 84.2042, rec_loss: 83.8565, KL_loss: 0.1540, MMD_loss: 0.1937 Epoch 296/300 Batch 0000/15, Loss: 84.8976, rec_loss: 84.5420, KL_loss: 0.1526, MMD_loss: 0.2030 Epoch 296/300 Batch 0015/15, Loss: 85.6030, rec_loss: 85.2593, KL_loss: 0.1507, MMD_loss: 0.1929 Epoch 297/300 Batch 0000/15, Loss: 85.1993, rec_loss: 84.9023, KL_loss: 0.1506, MMD_loss: 0.1464 Epoch 297/300 Batch 0015/15, Loss: 85.9585, rec_loss: 85.5340, KL_loss: 0.1534, MMD_loss: 0.2711 Epoch 298/300 Batch 0000/15, Loss: 82.7905, rec_loss: 82.4496, KL_loss: 0.1507, MMD_loss: 0.1902 Epoch 298/300 Batch 0015/15, Loss: 88.2119, rec_loss: 87.8518, KL_loss: 0.1466, MMD_loss: 0.2134 Epoch 299/300 Batch 0000/15, Loss: 87.1541, rec_loss: 86.7917, KL_loss: 0.1495, MMD_loss: 0.2128 Epoch 299/300 Batch 0015/15, Loss: 85.1953, rec_loss: 84.9000, KL_loss: 0.1506, MMD_loss: 0.1447
You can use MMD layer representation for batch-removal
c
can be the batch or the condition that you want your data to be mapped on it.
latent_y = model.get_y(
adata.X, c=model.label_encoder.transform(np.tile(np.array("Control"), len(adata))))
adata_latent = sc.AnnData(latent_y)
adata_latent.obs["cell_label"] = adata.obs["cell_label"].tolist()
adata_latent.obs[condition_key] = adata.obs[condition_key].tolist()
sc.pp.neighbors(adata_latent)
sc.tl.umap(adata_latent)
sc.pl.umap(adata_latent, color=[condition_key, "cell_label"])
... storing 'cell_label' as categorical ... storing 'condition' as categorical
ground_truth = adata[((adata.obs["cell_label"] == "TA")
& (adata.obs["condition"].isin(["Hpoly.Day10", "Control"])))]
adata_source = adata[(adata.obs["cell_label"] == "Stem") &
(adata.obs["condition"] == "Hpoly.Day10")]
predicted_data = model.predict(x=adata_source.X, y=adata_source.obs["condition"].tolist(),
target="Hpoly.Day10")
adata_pred = sc.AnnData(predicted_data)
adata_pred.obs["condition"] = np.tile("predicted", len(adata_pred))
adata_pred.var_names = adata_source.var_names.tolist()
all_adata = ground_truth.concatenate(adata_pred)
sc.tl.pca(all_adata)
sc.pl.pca(all_adata, color=["condition"])
sc.pl.violin(all_adata, keys="Reg3b", groupby="condition")
... storing 'barcode' as categorical ... storing 'cell_label' as categorical ... storing 'condition' as categorical
target
can be the batch or the condition that you want your data to be mapped on it.
batch_removed = model.predict(x=adata.X, y=adata.obs[condition_key].tolist(),
target="Control")
corrected = sc.AnnData(batch_removed)
corrected.obs["cell_label"] = adata.obs["cell_label"].tolist()
corrected.obs[condition_key] = adata.obs[condition_key].tolist()
sc.pp.neighbors(corrected)
sc.tl.umap(corrected)
WARNING: You’re trying to run this on 2000 dimensions of `.X`, if you really want this, set `use_rep='X'`. Falling back to preprocessing with `sc.pp.pca` and default params.
sc.pl.umap(corrected, color=[condition_key, "cell_label"])
... storing 'cell_label' as categorical ... storing 'condition' as categorical