In [3]:
#Load some packages and add workers
using Images
using Distributed

using LinearAlgebra
using Statistics
In [5]:
#Load the package on all workers
addprocs(4)
@everywhere using DPMMSubClusters
In [8]:
#Load image, taken from SINTEL http://sintel.is.tue.mpg.de/
img = load("frame_0001.png")
Out[8]:
In [9]:
#Change to channel view
img_channels = channelview(img)
z,x,y = size(img_channels)
Out[9]:
(3, 436, 1024)
In [10]:
#Create input
input_arr = zeros(5,x*y)
for i=1:x
    for j=1:y
        input_arr[:,(i-1)*y+j]=[img_channels[1,i,j],img_channels[2,i,j],img_channels[3,i,j],i,j]
    end
end
In [11]:
#Create HyperParams
#The rgb,xy multiplier allows us to both play with the weight of the xy/rgb
rgb_prior_multiplier = 30.0
xy_prior_multiplier = 1.0

data_cov = cov(input_arr')
data_cov[4:5,1:3] .= 0
data_cov[1:3,4:5] .= 0

data_cov[1:3,1:3] .*= rgb_prior_multiplier
data_cov[4:5,4:5] .*= xy_prior_multiplier

data_mean = mean(input_arr,dims = 2)[:]

hyper_params = DPMMSubClusters.niw_hyperparams(1.0,
           data_mean,
           8,
           data_cov)
Out[11]:
DPMMSubClusters.niw_hyperparams(1.0f0, Float32[0.10742735, 0.17671652, 0.18514434, 218.5, 512.5], 8.0f0, Float32[0.11475264 0.08273591 … 0.0 0.0; 0.08273591 0.11883843 … 0.0 0.0; … ; 0.0 0.0 … 15841.285 0.0; 0.0 0.0 … 0.0 87381.445])
In [14]:
#Run the model
labels,clusters,weights = DPMMSubClusters.fit(input_arr,hyper_params,1.0,iters = 300, verbose = true)
Loading and distributing data:
  0.005793 seconds (1.22 k allocations: 8.572 MiB)
Node Leaders:
Dict{Any,Any}(2 => Any[2, 3, 4, 5, 6, 7, 8, 9])
Iteration: 1 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07701611518859863 || Total time:0.07701611518859863
Iteration: 2 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08147692680358887 || Total time:0.1584930419921875
Iteration: 3 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07693290710449219 || Total time:0.2354259490966797
Iteration: 4 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07924199104309082 || Total time:0.3146679401397705
Iteration: 5 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07328391075134277 || Total time:0.3879518508911133
Iteration: 6 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07474493980407715 || Total time:0.46269679069519043
Iteration: 7 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07329201698303223 || Total time:0.5359888076782227
Iteration: 8 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08418893814086914 || Total time:0.6201777458190918
Iteration: 9 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07021784782409668 || Total time:0.6903955936431885
Iteration: 10 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07091999053955078 || Total time:0.7613155841827393
Iteration: 11 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.0753469467163086 || Total time:0.8366625308990479
Iteration: 12 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07974481582641602 || Total time:0.9164073467254639
Iteration: 13 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.06827712059020996 || Total time:0.9846844673156738
Iteration: 14 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.0800638198852539 || Total time:1.0647482872009277
Iteration: 15 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07579493522644043 || Total time:1.1405432224273682
Iteration: 16 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08474087715148926 || Total time:1.2252840995788574
Iteration: 17 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07435917854309082 || Total time:1.2996432781219482
Iteration: 18 || Clusters count: 1 || Log posterior: -2.94030621042512e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07466006278991699 || Total time:1.3743033409118652
Iteration: 19 || Clusters count: 2 || Log posterior: -2.433403032194835e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.09824681282043457 || Total time:1.4725501537322998
Iteration: 20 || Clusters count: 2 || Log posterior: -2.434035030668956e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08364200592041016 || Total time:1.55619215965271
Iteration: 21 || Clusters count: 2 || Log posterior: -2.434272159221393e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07743501663208008 || Total time:1.63362717628479
Iteration: 22 || Clusters count: 2 || Log posterior: -2.4336362816405124e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08759212493896484 || Total time:1.7212193012237549
Iteration: 23 || Clusters count: 2 || Log posterior: -2.4340017832021527e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.0835731029510498 || Total time:1.8047924041748047
Iteration: 24 || Clusters count: 2 || Log posterior: -2.433915407701473e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08720898628234863 || Total time:1.8920013904571533
Iteration: 25 || Clusters count: 2 || Log posterior: -2.433962658118706e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08809685707092285 || Total time:1.9800982475280762
Iteration: 26 || Clusters count: 2 || Log posterior: -2.433943410443287e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.09091615676879883 || Total time:2.071014404296875
Iteration: 27 || Clusters count: 2 || Log posterior: -2.433845157701473e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08568215370178223 || Total time:2.1566965579986572
Iteration: 28 || Clusters count: 2 || Log posterior: -2.434018532105428e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08578705787658691 || Total time:2.242483615875244
Iteration: 29 || Clusters count: 2 || Log posterior: -2.43401965739749e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08719491958618164 || Total time:2.329678535461426
Iteration: 30 || Clusters count: 2 || Log posterior: -2.433904406044465e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08775591850280762 || Total time:2.4174344539642334
Iteration: 31 || Clusters count: 2 || Log posterior: -2.434116407153111e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08286499977111816 || Total time:2.5002994537353516
Iteration: 32 || Clusters count: 2 || Log posterior: -2.4338197818491277e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08023381233215332 || Total time:2.580533266067505
Iteration: 33 || Clusters count: 2 || Log posterior: -2.433897784394245e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.09551620483398438 || Total time:2.6760494709014893
Iteration: 34 || Clusters count: 2 || Log posterior: -2.4334845346445846e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.0941619873046875 || Total time:2.7702114582061768
Iteration: 35 || Clusters count: 2 || Log posterior: -2.433811782999498e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.09938311576843262 || Total time:2.8695945739746094
Iteration: 36 || Clusters count: 2 || Log posterior: -2.433789781980257e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.07958602905273438 || Total time:2.9491806030273438
Iteration: 37 || Clusters count: 2 || Log posterior: -2.4342737835001773e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08339095115661621 || Total time:3.03257155418396
Iteration: 38 || Clusters count: 2 || Log posterior: -2.433920159561138e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.08975005149841309 || Total time:3.122321605682373
Iteration: 39 || Clusters count: 4 || Log posterior: -2.1547813297006898e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.14000201225280762 || Total time:3.2623236179351807
Iteration: 40 || Clusters count: 4 || Log posterior: -2.127503348286372e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.11730599403381348 || Total time:3.379629611968994
Iteration: 41 || Clusters count: 4 || Log posterior: -2.1208226351969056e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1163640022277832 || Total time:3.4959936141967773
Iteration: 42 || Clusters count: 4 || Log posterior: -2.1198485794255557e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.10796594619750977 || Total time:3.603959560394287
Iteration: 43 || Clusters count: 4 || Log posterior: -2.119565196038466e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.12027215957641602 || Total time:3.724231719970703
Iteration: 44 || Clusters count: 4 || Log posterior: -2.119063896709663e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1139059066772461 || Total time:3.838137626647949
Iteration: 45 || Clusters count: 4 || Log posterior: -2.1186015423731627e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.11861801147460938 || Total time:3.9567556381225586
Iteration: 46 || Clusters count: 4 || Log posterior: -2.1185462108738245e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1084749698638916 || Total time:4.06523060798645
Iteration: 47 || Clusters count: 4 || Log posterior: -2.11828393929432e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.10605192184448242 || Total time:4.171282529830933
Iteration: 48 || Clusters count: 4 || Log posterior: -2.118084934603434e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.10657095909118652 || Total time:4.277853488922119
Iteration: 49 || Clusters count: 4 || Log posterior: -2.1180476747693834e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.11624312400817871 || Total time:4.394096612930298
Iteration: 50 || Clusters count: 4 || Log posterior: -2.1181455573742203e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.12472414970397949 || Total time:4.518820762634277
Iteration: 51 || Clusters count: 4 || Log posterior: -2.1182880160867986e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.09134602546691895 || Total time:4.610166788101196
Iteration: 52 || Clusters count: 4 || Log posterior: -2.1185766340176873e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.09686112403869629 || Total time:4.707027912139893
Iteration: 53 || Clusters count: 4 || Log posterior: -2.1178688458859734e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1057131290435791 || Total time:4.812741041183472
Iteration: 54 || Clusters count: 4 || Log posterior: -2.1179034586438946e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.09629297256469727 || Total time:4.909034013748169
Iteration: 55 || Clusters count: 4 || Log posterior: -2.1183496708724317e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.09841489791870117 || Total time:5.00744891166687
Iteration: 56 || Clusters count: 4 || Log posterior: -2.118578444872599e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.09620308876037598 || Total time:5.103652000427246
Iteration: 57 || Clusters count: 4 || Log posterior: -2.117961105120641e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.10171890258789062 || Total time:5.205370903015137
Iteration: 58 || Clusters count: 4 || Log posterior: -2.1185446714532194e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.09511303901672363 || Total time:5.30048394203186
Iteration: 59 || Clusters count: 8 || Log posterior: -1.9542992851860346e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.12170290946960449 || Total time:5.422186851501465
Iteration: 60 || Clusters count: 8 || Log posterior: -1.9319600557374773e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.12904810905456543 || Total time:5.55123496055603
Iteration: 61 || Clusters count: 8 || Log posterior: -1.9132618928360757e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.14588618278503418 || Total time:5.6971211433410645
Iteration: 62 || Clusters count: 8 || Log posterior: -1.8975245592813303e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1505589485168457 || Total time:5.84768009185791
Iteration: 63 || Clusters count: 8 || Log posterior: -1.887646973384839e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13890719413757324 || Total time:5.986587285995483
Iteration: 64 || Clusters count: 8 || Log posterior: -1.88345448942421e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13276982307434082 || Total time:6.119357109069824
Iteration: 65 || Clusters count: 8 || Log posterior: -1.8828814948544325e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13953113555908203 || Total time:6.258888244628906
Iteration: 66 || Clusters count: 8 || Log posterior: -1.8821407711176686e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13500213623046875 || Total time:6.393890380859375
Iteration: 67 || Clusters count: 8 || Log posterior: -1.8818504211752224e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1356971263885498 || Total time:6.529587507247925
Iteration: 68 || Clusters count: 8 || Log posterior: -1.881837356393795e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13566899299621582 || Total time:6.665256500244141
Iteration: 69 || Clusters count: 8 || Log posterior: -1.8814452477571783e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.12687110900878906 || Total time:6.79212760925293
Iteration: 70 || Clusters count: 8 || Log posterior: -1.88109204112075e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13169097900390625 || Total time:6.923818588256836
Iteration: 71 || Clusters count: 8 || Log posterior: -1.8805551569742968e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1318988800048828 || Total time:7.055717468261719
Iteration: 72 || Clusters count: 8 || Log posterior: -1.880696114725572e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1374499797821045 || Total time:7.193167448043823
Iteration: 73 || Clusters count: 8 || Log posterior: -1.8805658208894555e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13457202911376953 || Total time:7.327739477157593
Iteration: 74 || Clusters count: 8 || Log posterior: -1.8801205057067676e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13284683227539062 || Total time:7.460586309432983
Iteration: 75 || Clusters count: 8 || Log posterior: -1.8798441152908618e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13097596168518066 || Total time:7.591562271118164
Iteration: 76 || Clusters count: 8 || Log posterior: -1.8790968911070647e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13871097564697266 || Total time:7.730273246765137
Iteration: 77 || Clusters count: 8 || Log posterior: -1.8792325066873853e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.14544415473937988 || Total time:7.875717401504517
Iteration: 78 || Clusters count: 8 || Log posterior: -1.8791880708849258e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.13271093368530273 || Total time:8.00842833518982
Iteration: 79 || Clusters count: 13 || Log posterior: -1.846167093545419e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.15891504287719727 || Total time:8.167343378067017
Iteration: 80 || Clusters count: 13 || Log posterior: -1.828759635206681e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1755998134613037 || Total time:8.34294319152832
Iteration: 81 || Clusters count: 13 || Log posterior: -1.819487275620443e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1774280071258545 || Total time:8.520371198654175
Iteration: 82 || Clusters count: 13 || Log posterior: -1.8118097939038093e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.18114209175109863 || Total time:8.701513290405273
Iteration: 83 || Clusters count: 13 || Log posterior: -1.8026219297361195e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1934499740600586 || Total time:8.894963264465332
Iteration: 84 || Clusters count: 13 || Log posterior: -1.793482851239901e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.17632818222045898 || Total time:9.071291446685791
Iteration: 85 || Clusters count: 13 || Log posterior: -1.7869504681980438e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1771230697631836 || Total time:9.248414516448975
Iteration: 86 || Clusters count: 13 || Log posterior: -1.783496927559596e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1808779239654541 || Total time:9.429292440414429
Iteration: 87 || Clusters count: 13 || Log posterior: -1.781680208748562e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.18771100044250488 || Total time:9.617003440856934
Iteration: 88 || Clusters count: 14 || Log posterior: -1.77916405354498e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.18670296669006348 || Total time:9.803706407546997
Iteration: 89 || Clusters count: 14 || Log posterior: -1.771186363479596e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2343769073486328 || Total time:10.03808331489563
Iteration: 90 || Clusters count: 14 || Log posterior: -1.7592191805784525e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.18394112586975098 || Total time:10.22202444076538
Iteration: 91 || Clusters count: 14 || Log posterior: -1.7524464136812496e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2024211883544922 || Total time:10.424445629119873
Iteration: 92 || Clusters count: 14 || Log posterior: -1.747775424565535e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1826341152191162 || Total time:10.60707974433899
Iteration: 93 || Clusters count: 14 || Log posterior: -1.7454810647675805e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.18200087547302246 || Total time:10.789080619812012
Iteration: 94 || Clusters count: 14 || Log posterior: -1.7438572659344487e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.18517708778381348 || Total time:10.974257707595825
Iteration: 95 || Clusters count: 14 || Log posterior: -1.7420037774770078e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.1880779266357422 || Total time:11.162335634231567
Iteration: 96 || Clusters count: 14 || Log posterior: -1.738403822297317e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2765669822692871 || Total time:11.438902616500854
Iteration: 97 || Clusters count: 14 || Log posterior: -1.7349330810713586e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.21088790893554688 || Total time:11.649790525436401
Iteration: 98 || Clusters count: 14 || Log posterior: -1.7314633156678495e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.19324707984924316 || Total time:11.843037605285645
Iteration: 99 || Clusters count: 17 || Log posterior: -1.7183199848562297e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.20221900939941406 || Total time:12.045256614685059
Iteration: 100 || Clusters count: 17 || Log posterior: -1.7082228726428566e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.21535801887512207 || Total time:12.26061463356018
Iteration: 101 || Clusters count: 19 || Log posterior: -1.6995967491306476e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.22843480110168457 || Total time:12.489049434661865
Iteration: 102 || Clusters count: 20 || Log posterior: -1.6952213825565395e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3155357837677002 || Total time:12.804585218429565
Iteration: 103 || Clusters count: 20 || Log posterior: -1.6907343624825291e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34720802307128906 || Total time:13.151793241500854
Iteration: 104 || Clusters count: 20 || Log posterior: -1.6885413643542337e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.22956395149230957 || Total time:13.381357192993164
Iteration: 105 || Clusters count: 20 || Log posterior: -1.6875561995031638e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.23163509368896484 || Total time:13.612992286682129
Iteration: 106 || Clusters count: 20 || Log posterior: -1.6865375626753392e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.23239994049072266 || Total time:13.845392227172852
Iteration: 107 || Clusters count: 20 || Log posterior: -1.685842281089764e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.23799586296081543 || Total time:14.083388090133667
Iteration: 108 || Clusters count: 20 || Log posterior: -1.6846200349008855e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.30910491943359375 || Total time:14.39249300956726
Iteration: 109 || Clusters count: 20 || Log posterior: -1.6840397815609989e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.24216008186340332 || Total time:14.634653091430664
Iteration: 110 || Clusters count: 20 || Log posterior: -1.6833641619924125e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.30151915550231934 || Total time:14.936172246932983
Iteration: 111 || Clusters count: 20 || Log posterior: -1.682533678605181e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.29640698432922363 || Total time:15.232579231262207
Iteration: 112 || Clusters count: 20 || Log posterior: -1.682184170188289e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2806110382080078 || Total time:15.513190269470215
Iteration: 113 || Clusters count: 20 || Log posterior: -1.6816469898471655e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.282944917678833 || Total time:15.796135187149048
Iteration: 114 || Clusters count: 20 || Log posterior: -1.681070056340677e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.22408199310302734 || Total time:16.020217180252075
Iteration: 115 || Clusters count: 20 || Log posterior: -1.6806079388276143e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2800769805908203 || Total time:16.300294160842896
Iteration: 116 || Clusters count: 20 || Log posterior: -1.6805168742435987e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2976088523864746 || Total time:16.59790301322937
Iteration: 117 || Clusters count: 20 || Log posterior: -1.6799909684425406e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3008279800415039 || Total time:16.898730993270874
Iteration: 118 || Clusters count: 20 || Log posterior: -1.6808604604812681e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34835195541381836 || Total time:17.247082948684692
Iteration: 119 || Clusters count: 20 || Log posterior: -1.6803343296774435e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.22816801071166992 || Total time:17.475250959396362
Iteration: 120 || Clusters count: 20 || Log posterior: -1.6805759493486704e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.27920985221862793 || Total time:17.75446081161499
Iteration: 121 || Clusters count: 21 || Log posterior: -1.6798088212761686e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2503988742828369 || Total time:18.004859685897827
Iteration: 122 || Clusters count: 23 || Log posterior: -1.6763203521054806e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.35361599922180176 || Total time:18.35847568511963
Iteration: 123 || Clusters count: 23 || Log posterior: -1.672466299737674e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.26455211639404297 || Total time:18.623027801513672
Iteration: 124 || Clusters count: 23 || Log posterior: -1.670719880836588e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.30954694747924805 || Total time:18.93257474899292
Iteration: 125 || Clusters count: 23 || Log posterior: -1.670131963242394e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.33812808990478516 || Total time:19.270702838897705
Iteration: 126 || Clusters count: 23 || Log posterior: -1.6692036365786623e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2427361011505127 || Total time:19.513438940048218
Iteration: 127 || Clusters count: 23 || Log posterior: -1.6687724413685617e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.31324005126953125 || Total time:19.82667899131775
Iteration: 128 || Clusters count: 23 || Log posterior: -1.6688052744266812e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.24716496467590332 || Total time:20.073843955993652
Iteration: 129 || Clusters count: 23 || Log posterior: -1.6682347806404638e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.33892202377319336 || Total time:20.412765979766846
Iteration: 130 || Clusters count: 23 || Log posterior: -1.6684460488818707e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.25061798095703125 || Total time:20.663383960723877
Iteration: 131 || Clusters count: 23 || Log posterior: -1.6678074026963527e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.32848310470581055 || Total time:20.991867065429688
Iteration: 132 || Clusters count: 23 || Log posterior: -1.667948782680254e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3107490539550781 || Total time:21.302616119384766
Iteration: 133 || Clusters count: 23 || Log posterior: -1.6681963470343165e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2982048988342285 || Total time:21.600821018218994
Iteration: 134 || Clusters count: 24 || Log posterior: -1.668052414553385e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2779378890991211 || Total time:21.878758907318115
Iteration: 135 || Clusters count: 24 || Log posterior: -1.6666241774483747e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.35571885108947754 || Total time:22.234477758407593
Iteration: 136 || Clusters count: 24 || Log posterior: -1.665687594253164e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3291330337524414 || Total time:22.563610792160034
Iteration: 137 || Clusters count: 24 || Log posterior: -1.6649984945139699e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3237330913543701 || Total time:22.887343883514404
Iteration: 138 || Clusters count: 24 || Log posterior: -1.6652440202293214e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.4149510860443115 || Total time:23.302294969558716
Iteration: 139 || Clusters count: 24 || Log posterior: -1.6655695465982952e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.32305097579956055 || Total time:23.625345945358276
Iteration: 140 || Clusters count: 24 || Log posterior: -1.6650935106271326e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2747519016265869 || Total time:23.900097846984863
Iteration: 141 || Clusters count: 24 || Log posterior: -1.6657128600323023e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.31745290756225586 || Total time:24.21755075454712
Iteration: 142 || Clusters count: 24 || Log posterior: -1.6650246801456031e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.25440311431884766 || Total time:24.471953868865967
Iteration: 143 || Clusters count: 24 || Log posterior: -1.6650183616154017e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.33341121673583984 || Total time:24.805365085601807
Iteration: 144 || Clusters count: 25 || Log posterior: -1.6647140063378622e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.33281779289245605 || Total time:25.138182878494263
Iteration: 145 || Clusters count: 25 || Log posterior: -1.6597348198150224e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.38100194931030273 || Total time:25.519184827804565
Iteration: 146 || Clusters count: 25 || Log posterior: -1.6583352159011187e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.25686097145080566 || Total time:25.77604579925537
Iteration: 147 || Clusters count: 25 || Log posterior: -1.6573634535101468e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.30566906929016113 || Total time:26.081714868545532
Iteration: 148 || Clusters count: 25 || Log posterior: -1.6575171220427558e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.26935791969299316 || Total time:26.351072788238525
Iteration: 149 || Clusters count: 25 || Log posterior: -1.6564630520911026e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3713710308074951 || Total time:26.72244381904602
Iteration: 150 || Clusters count: 25 || Log posterior: -1.657039741504889e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34081006050109863 || Total time:27.06325387954712
Iteration: 151 || Clusters count: 25 || Log posterior: -1.656698766299348e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3355419635772705 || Total time:27.39879584312439
Iteration: 152 || Clusters count: 25 || Log posterior: -1.6565757830185718e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2612581253051758 || Total time:27.660053968429565
Iteration: 153 || Clusters count: 25 || Log posterior: -1.6567045762504153e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.25799012184143066 || Total time:27.918044090270996
Iteration: 154 || Clusters count: 25 || Log posterior: -1.657010466557008e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.32255101203918457 || Total time:28.24059510231018
Iteration: 155 || Clusters count: 25 || Log posterior: -1.6563569487054169e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.361907958984375 || Total time:28.602503061294556
Iteration: 156 || Clusters count: 25 || Log posterior: -1.6570249643155157e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.40380215644836426 || Total time:29.00630521774292
Iteration: 157 || Clusters count: 25 || Log posterior: -1.6561723117472942e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.36090612411499023 || Total time:29.36721134185791
Iteration: 158 || Clusters count: 25 || Log posterior: -1.6560615813671171e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2578740119934082 || Total time:29.62508535385132
Iteration: 159 || Clusters count: 25 || Log posterior: -1.6564637247035326e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.26338720321655273 || Total time:29.88847255706787
Iteration: 160 || Clusters count: 25 || Log posterior: -1.6563615100790085e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2961549758911133 || Total time:30.184627532958984
Iteration: 161 || Clusters count: 25 || Log posterior: -1.6560575973069477e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3741309642791748 || Total time:30.55875849723816
Iteration: 162 || Clusters count: 25 || Log posterior: -1.656091422452551e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3630368709564209 || Total time:30.92179536819458
Iteration: 163 || Clusters count: 25 || Log posterior: -1.656233705208997e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3901069164276123 || Total time:31.311902284622192
Iteration: 164 || Clusters count: 25 || Log posterior: -1.656446095453722e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2931499481201172 || Total time:31.60505223274231
Iteration: 165 || Clusters count: 25 || Log posterior: -1.6560618058938808e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.29768800735473633 || Total time:31.902740240097046
Iteration: 166 || Clusters count: 25 || Log posterior: -1.6558875429247438e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3573930263519287 || Total time:32.260133266448975
Iteration: 167 || Clusters count: 25 || Log posterior: -1.6561199337593145e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.35101985931396484 || Total time:32.61115312576294
Iteration: 168 || Clusters count: 25 || Log posterior: -1.656434800674419e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3278989791870117 || Total time:32.93905210494995
Iteration: 169 || Clusters count: 25 || Log posterior: -1.656043150072795e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34340691566467285 || Total time:33.282459020614624
Iteration: 170 || Clusters count: 25 || Log posterior: -1.6557527484488299e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2735579013824463 || Total time:33.55601692199707
Iteration: 171 || Clusters count: 25 || Log posterior: -1.6555112417826483e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.26056694984436035 || Total time:33.81658387184143
Iteration: 172 || Clusters count: 25 || Log posterior: -1.6560670735735698e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.37394285202026367 || Total time:34.190526723861694
Iteration: 173 || Clusters count: 25 || Log posterior: -1.6553693036604945e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34495997428894043 || Total time:34.535486698150635
Iteration: 174 || Clusters count: 26 || Log posterior: -1.6548848293286392e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.33490705490112305 || Total time:34.87039375305176
Iteration: 175 || Clusters count: 26 || Log posterior: -1.6529419969640782e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3388841152191162 || Total time:35.209277868270874
Iteration: 176 || Clusters count: 26 || Log posterior: -1.6512569643215947e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.28012895584106445 || Total time:35.48940682411194
Iteration: 177 || Clusters count: 26 || Log posterior: -1.6508852375309272e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2754650115966797 || Total time:35.76487183570862
Iteration: 178 || Clusters count: 26 || Log posterior: -1.650446214168769e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34834885597229004 || Total time:36.11322069168091
Iteration: 179 || Clusters count: 26 || Log posterior: -1.6502752220096388e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.399137020111084 || Total time:36.51235771179199
Iteration: 180 || Clusters count: 26 || Log posterior: -1.6501093263304043e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3948659896850586 || Total time:36.90722370147705
Iteration: 181 || Clusters count: 26 || Log posterior: -1.6497562227878382e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.40567708015441895 || Total time:37.31290078163147
Iteration: 182 || Clusters count: 26 || Log posterior: -1.6509702628829298e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3214879035949707 || Total time:37.63438868522644
Iteration: 183 || Clusters count: 26 || Log posterior: -1.650338316171747e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3087887763977051 || Total time:37.943177461624146
Iteration: 184 || Clusters count: 26 || Log posterior: -1.650537652562479e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.38761210441589355 || Total time:38.33078956604004
Iteration: 185 || Clusters count: 26 || Log posterior: -1.6505934528971745e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.41709208488464355 || Total time:38.74788165092468
Iteration: 186 || Clusters count: 26 || Log posterior: -1.6504939520764141e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.4523758888244629 || Total time:39.200257539749146
Iteration: 187 || Clusters count: 26 || Log posterior: -1.6506793597389273e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3771839141845703 || Total time:39.577441453933716
Iteration: 188 || Clusters count: 26 || Log posterior: -1.6507253251942215e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34453415870666504 || Total time:39.92197561264038
Iteration: 189 || Clusters count: 26 || Log posterior: -1.6506406823916251e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3258509635925293 || Total time:40.24782657623291
Iteration: 190 || Clusters count: 26 || Log posterior: -1.6503747768908555e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3479020595550537 || Total time:40.595728635787964
Iteration: 191 || Clusters count: 26 || Log posterior: -1.6507685104608359e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3777928352355957 || Total time:40.97352147102356
Iteration: 192 || Clusters count: 26 || Log posterior: -1.6508358390362076e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.38831520080566406 || Total time:41.361836671829224
Iteration: 193 || Clusters count: 26 || Log posterior: -1.650528848055344e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.4105260372161865 || Total time:41.77236270904541
Iteration: 194 || Clusters count: 26 || Log posterior: -1.6502489860436732e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.29678988456726074 || Total time:42.06915259361267
Iteration: 195 || Clusters count: 26 || Log posterior: -1.6497977957469046e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.30745601654052734 || Total time:42.3766086101532
Iteration: 196 || Clusters count: 26 || Log posterior: -1.6495853009147458e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3857579231262207 || Total time:42.76236653327942
Iteration: 197 || Clusters count: 26 || Log posterior: -1.6502539481253447e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.44280099868774414 || Total time:43.20516753196716
Iteration: 198 || Clusters count: 26 || Log posterior: -1.6500836471174774e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.4062788486480713 || Total time:43.611446380615234
Iteration: 199 || Clusters count: 26 || Log posterior: -1.6496267128914422e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.41059303283691406 || Total time:44.02203941345215
Iteration: 200 || Clusters count: 26 || Log posterior: -1.6494094036686227e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2987029552459717 || Total time:44.32074236869812
Iteration: 201 || Clusters count: 26 || Log posterior: -1.6494136544954597e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34648704528808594 || Total time:44.667229413986206
Iteration: 202 || Clusters count: 26 || Log posterior: -1.649521501218778e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3994138240814209 || Total time:45.06664323806763
Iteration: 203 || Clusters count: 26 || Log posterior: -1.6488647973875806e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3461441993713379 || Total time:45.412787437438965
Iteration: 204 || Clusters count: 26 || Log posterior: -1.649197438083869e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3800170421600342 || Total time:45.792804479599
Iteration: 205 || Clusters count: 26 || Log posterior: -1.6495019069036082e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3486309051513672 || Total time:46.141435384750366
Iteration: 206 || Clusters count: 26 || Log posterior: -1.6489037847263634e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2859830856323242 || Total time:46.42741847038269
Iteration: 207 || Clusters count: 26 || Log posterior: -1.6489492359857385e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.28516602516174316 || Total time:46.712584495544434
Iteration: 208 || Clusters count: 26 || Log posterior: -1.649213879825693e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.32728099822998047 || Total time:47.039865493774414
Iteration: 209 || Clusters count: 26 || Log posterior: -1.648753704101306e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34081077575683594 || Total time:47.38067626953125
Iteration: 210 || Clusters count: 26 || Log posterior: -1.648731759451013e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.32782697677612305 || Total time:47.70850324630737
Iteration: 211 || Clusters count: 26 || Log posterior: -1.6484357354482224e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3861360549926758 || Total time:48.09463930130005
Iteration: 212 || Clusters count: 27 || Log posterior: -1.6487049309036564e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2781028747558594 || Total time:48.37274217605591
Iteration: 213 || Clusters count: 26 || Log posterior: -1.648791998271327e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.30341196060180664 || Total time:48.676154136657715
Iteration: 214 || Clusters count: 26 || Log posterior: -1.6489838020463996e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.33722782135009766 || Total time:49.01338195800781
Iteration: 215 || Clusters count: 26 || Log posterior: -1.648945324190002e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3988049030303955 || Total time:49.41218686103821
Iteration: 216 || Clusters count: 26 || Log posterior: -1.648824830096942e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3330090045928955 || Total time:49.7451958656311
Iteration: 217 || Clusters count: 26 || Log posterior: -1.6488230611238307e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.37682414054870605 || Total time:50.12202000617981
Iteration: 218 || Clusters count: 26 || Log posterior: -1.6486906463526313e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2701871395111084 || Total time:50.39220714569092
Iteration: 219 || Clusters count: 26 || Log posterior: -1.648779684233409e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2727620601654053 || Total time:50.66496920585632
Iteration: 220 || Clusters count: 26 || Log posterior: -1.6492405817828015e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.36339521408081055 || Total time:51.028364419937134
Iteration: 221 || Clusters count: 26 || Log posterior: -1.6489612058378279e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3312249183654785 || Total time:51.35958933830261
Iteration: 222 || Clusters count: 26 || Log posterior: -1.6495908196309626e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34119606018066406 || Total time:51.700785398483276
Iteration: 223 || Clusters count: 26 || Log posterior: -1.6490471389134934e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3919999599456787 || Total time:52.092785358428955
Iteration: 224 || Clusters count: 26 || Log posterior: -1.6489947209465324e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2763528823852539 || Total time:52.36913824081421
Iteration: 225 || Clusters count: 26 || Log posterior: -1.6485914585767798e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2802550792694092 || Total time:52.64939332008362
Iteration: 226 || Clusters count: 26 || Log posterior: -1.6487117337363772e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3437049388885498 || Total time:52.99309825897217
Iteration: 227 || Clusters count: 26 || Log posterior: -1.6489537507891469e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3384819030761719 || Total time:53.33158016204834
Iteration: 228 || Clusters count: 26 || Log posterior: -1.6495987741204312e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34424901008605957 || Total time:53.6758291721344
Iteration: 229 || Clusters count: 26 || Log posterior: -1.648675981807452e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.37922000885009766 || Total time:54.0550491809845
Iteration: 230 || Clusters count: 26 || Log posterior: -1.6492659813821132e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.28719592094421387 || Total time:54.34224510192871
Iteration: 231 || Clusters count: 26 || Log posterior: -1.6490488479862027e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.27440810203552246 || Total time:54.61665320396423
Iteration: 232 || Clusters count: 26 || Log posterior: -1.6489385716855335e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3512840270996094 || Total time:54.96793723106384
Iteration: 233 || Clusters count: 26 || Log posterior: -1.6500339604804327e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3251688480377197 || Total time:55.29310607910156
Iteration: 234 || Clusters count: 26 || Log posterior: -1.6490434085939939e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3230600357055664 || Total time:55.61616611480713
Iteration: 235 || Clusters count: 26 || Log posterior: -1.6493592689890673e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.35462212562561035 || Total time:55.97078824043274
Iteration: 236 || Clusters count: 26 || Log posterior: -1.6494289398140726e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.32036805152893066 || Total time:56.29115629196167
Iteration: 237 || Clusters count: 26 || Log posterior: -1.6486839165769632e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.270021915435791 || Total time:56.56117820739746
Iteration: 238 || Clusters count: 26 || Log posterior: -1.6487604432235775e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3598489761352539 || Total time:56.921027183532715
Iteration: 239 || Clusters count: 26 || Log posterior: -1.649142161486368e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.35803794860839844 || Total time:57.27906513214111
Iteration: 240 || Clusters count: 26 || Log posterior: -1.648942652774553e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3289620876312256 || Total time:57.60802721977234
Iteration: 241 || Clusters count: 26 || Log posterior: -1.6491711848602113e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.33690905570983887 || Total time:57.94493627548218
Iteration: 242 || Clusters count: 26 || Log posterior: -1.6496367970828826e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3400239944458008 || Total time:58.28496026992798
Iteration: 243 || Clusters count: 26 || Log posterior: -1.6488664306106388e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2928340435028076 || Total time:58.577794313430786
Iteration: 244 || Clusters count: 26 || Log posterior: -1.6491661674121432e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3417057991027832 || Total time:58.91950011253357
Iteration: 245 || Clusters count: 26 || Log posterior: -1.649213930142503e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3347201347351074 || Total time:59.25422024726868
Iteration: 246 || Clusters count: 26 || Log posterior: -1.6480046971465172e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3307621479034424 || Total time:59.58498239517212
Iteration: 247 || Clusters count: 26 || Log posterior: -1.6490465511952443e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3564457893371582 || Total time:59.94142818450928
Iteration: 248 || Clusters count: 26 || Log posterior: -1.6480195111579713e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3535580635070801 || Total time:60.29498624801636
Iteration: 249 || Clusters count: 26 || Log posterior: -1.6492786223338656e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.28494811058044434 || Total time:60.5799343585968
Iteration: 250 || Clusters count: 26 || Log posterior: -1.6489779565261416e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34063291549682617 || Total time:60.92056727409363
Iteration: 251 || Clusters count: 26 || Log posterior: -1.6486344773114724e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3341388702392578 || Total time:61.254706144332886
Iteration: 252 || Clusters count: 26 || Log posterior: -1.649225794279914e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3446049690246582 || Total time:61.599311113357544
Iteration: 253 || Clusters count: 26 || Log posterior: -1.649089449885827e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3680438995361328 || Total time:61.96735501289368
Iteration: 254 || Clusters count: 26 || Log posterior: -1.649535954661946e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34445691108703613 || Total time:62.31181192398071
Iteration: 255 || Clusters count: 26 || Log posterior: -1.6491015837619123e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2750821113586426 || Total time:62.586894035339355
Iteration: 256 || Clusters count: 26 || Log posterior: -1.6495325052277846e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3442261219024658 || Total time:62.93112015724182
Iteration: 257 || Clusters count: 26 || Log posterior: -1.648994998960714e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.31327199935913086 || Total time:63.24439215660095
Iteration: 258 || Clusters count: 26 || Log posterior: -1.6486594096780831e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34398698806762695 || Total time:63.58837914466858
Iteration: 259 || Clusters count: 26 || Log posterior: -1.6493150084807687e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34201478958129883 || Total time:63.93039393424988
Iteration: 260 || Clusters count: 26 || Log posterior: -1.6498777945104663e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3263070583343506 || Total time:64.25670099258423
Iteration: 261 || Clusters count: 26 || Log posterior: -1.6490537670839832e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3399820327758789 || Total time:64.59668302536011
Iteration: 262 || Clusters count: 26 || Log posterior: -1.6491654869538355e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.33322978019714355 || Total time:64.92991280555725
Iteration: 263 || Clusters count: 26 || Log posterior: -1.6494264087382373e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3111422061920166 || Total time:65.24105501174927
Iteration: 264 || Clusters count: 26 || Log posterior: -1.6496661260980186e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3233060836791992 || Total time:65.56436109542847
Iteration: 265 || Clusters count: 26 || Log posterior: -1.6490968117030663e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.32070493698120117 || Total time:65.88506603240967
Iteration: 266 || Clusters count: 26 || Log posterior: -1.6490225030424413e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.27071595191955566 || Total time:66.15578198432922
Iteration: 267 || Clusters count: 26 || Log posterior: -1.6488442582100453e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3368101119995117 || Total time:66.49259209632874
Iteration: 268 || Clusters count: 26 || Log posterior: -1.6489212413809118e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3537018299102783 || Total time:66.84629392623901
Iteration: 269 || Clusters count: 26 || Log posterior: -1.6494768646616759e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.32949304580688477 || Total time:67.1757869720459
Iteration: 270 || Clusters count: 27 || Log posterior: -1.6488233058426199e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3388550281524658 || Total time:67.51464200019836
Iteration: 271 || Clusters count: 27 || Log posterior: -1.649230132303936e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.35283708572387695 || Total time:67.86747908592224
Iteration: 272 || Clusters count: 27 || Log posterior: -1.648847571592074e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.28909897804260254 || Total time:68.15657806396484
Iteration: 273 || Clusters count: 27 || Log posterior: -1.648639296755177e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.37595701217651367 || Total time:68.53253507614136
Iteration: 274 || Clusters count: 27 || Log posterior: -1.648756874670606e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3394639492034912 || Total time:68.87199902534485
Iteration: 275 || Clusters count: 27 || Log posterior: -1.6488435379313054e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3703479766845703 || Total time:69.24234700202942
Iteration: 276 || Clusters count: 26 || Log posterior: -1.6487334996919455e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.27933406829833984 || Total time:69.52168107032776
Iteration: 277 || Clusters count: 26 || Log posterior: -1.6487075564520173e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3461740016937256 || Total time:69.86785507202148
Iteration: 278 || Clusters count: 26 || Log posterior: -1.6491637507877164e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.27666187286376953 || Total time:70.14451694488525
Iteration: 279 || Clusters count: 26 || Log posterior: -1.6491107889609162e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3628249168395996 || Total time:70.50734186172485
Iteration: 280 || Clusters count: 26 || Log posterior: -1.6488833370520878e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.311276912689209 || Total time:70.81861877441406
Iteration: 281 || Clusters count: 26 || Log posterior: -1.649273886126023e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34548115730285645 || Total time:71.16409993171692
Iteration: 282 || Clusters count: 26 || Log posterior: -1.648991621457915e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2929720878601074 || Total time:71.45707201957703
Iteration: 283 || Clusters count: 26 || Log posterior: -1.6485471376613195e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3344399929046631 || Total time:71.79151201248169
Iteration: 284 || Clusters count: 26 || Log posterior: -1.649037111099106e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.268967866897583 || Total time:72.06047987937927
Iteration: 285 || Clusters count: 26 || Log posterior: -1.649154241237145e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3506588935852051 || Total time:72.41113877296448
Iteration: 286 || Clusters count: 26 || Log posterior: -1.6493485169170904e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3692479133605957 || Total time:72.78038668632507
Iteration: 287 || Clusters count: 26 || Log posterior: -1.6493449645101856e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.4091649055480957 || Total time:73.18955159187317
Iteration: 288 || Clusters count: 27 || Log posterior: -1.6489389421241102e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2885918617248535 || Total time:73.47814345359802
Iteration: 289 || Clusters count: 27 || Log posterior: -1.6487032217258026e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.35358285903930664 || Total time:73.83172631263733
Iteration: 290 || Clusters count: 27 || Log posterior: -1.6480422081605021e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.28295016288757324 || Total time:74.1146764755249
Iteration: 291 || Clusters count: 27 || Log posterior: -1.6481432496118373e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34993696212768555 || Total time:74.46461343765259
Iteration: 292 || Clusters count: 27 || Log posterior: -1.6480765964690265e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3549342155456543 || Total time:74.81954765319824
Iteration: 293 || Clusters count: 27 || Log posterior: -1.6474840634485297e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3411440849304199 || Total time:75.16069173812866
Iteration: 294 || Clusters count: 27 || Log posterior: -1.6474974016742532e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2701249122619629 || Total time:75.43081665039062
Iteration: 295 || Clusters count: 27 || Log posterior: -1.6480412548153214e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.35790300369262695 || Total time:75.78871965408325
Iteration: 296 || Clusters count: 27 || Log posterior: -1.647426137575488e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.2731451988220215 || Total time:76.06186485290527
Iteration: 297 || Clusters count: 27 || Log posterior: -1.6471668919994659e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.3360021114349365 || Total time:76.39786696434021
Iteration: 298 || Clusters count: 27 || Log posterior: -1.6477608494939622e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.36333703994750977 || Total time:76.76120400428772
Iteration: 299 || Clusters count: 27 || Log posterior: -1.6485810737556033e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.34795594215393066 || Total time:77.10915994644165
Iteration: 300 || Clusters count: 27 || Log posterior: -1.6120752892685952e6 || Vi score: no gt || NMI score: no gt || Iter Time:0.29085397720336914 || Total time:77.40001392364502
Out[14]:
([26, 26, 26, 26, 26, 26, 26, 26, 26, 26  …  1, 1, 1, 1, 1, 1, 1, 1, 1, 1], DPMMSubClusters.mv_gaussian[DPMMSubClusters.mv_gaussian(Float32[0.08791604, 0.15855435, 0.16756245, 348.6046, 605.56934], Float32[0.00059554446 0.0009552731 … -0.054339442 -0.14335035; 0.0009552731 0.0015952496 … -0.40924647 -0.67706734; … ; -0.054339442 -0.40924647 … 2656.4346 2909.5305; -0.14335035 -0.67706734 … 2909.5305 56099.797], Float32[110213.445 -84754.97 … -7.719583 -0.104402415; -84754.97 270770.88 … 1.2206929 -0.109386705; … ; -7.719583 1.2206929 … 0.0010674179 -8.4442545e-6; -0.104402415 -0.109386705 … -8.4442545e-6 1.919085e-5], -11.481257f0, [331.98409968497674 -255.29827008690847 … -0.02325286961458397 -0.00031448016071472144; 0.0 453.42436476541326 … -0.010400245163921715 -0.00041831220505142067; … ; 0.0 0.0 … 0.01997789222546669 -0.0010361229368419215; 0.0 0.0 … 0.0 0.00422201099338806]), DPMMSubClusters.mv_gaussian(Float32[0.14800808, 0.13678575, 0.10723315, 212.15039, 466.4791], Float32[0.002676467 0.0025030752 … 0.6044567 0.98189133; 0.0025030752 0.002598142 … 0.6092531 1.0691746; … ; 0.6044567 0.6092531 … 848.6922 566.59735; 0.98189133 1.0691746 … 566.59735 1234.8599], Float32[5604.766 -11184.442 … 0.19732098 0.76166517; -11184.442 35509.496 … -1.7392819 -2.6136847; … ; 0.19732098 -1.7392819 … 0.0018491114 -0.00059727975; 0.76166517 -2.6136847 … -0.00059727975 0.0016647095], -11.395526f0, [74.86498491420642 -149.39484054510035 … 0.002635691276956154 0.010173850906256146; 0.0 114.85066999112904 … -0.011715414402162927 -0.00952335601360485; … ; 0.0 0.0 … 0.04121417442116419 -0.018910519586499558; 0.0 0.0 … 0.0 0.028457133624812236]), DPMMSubClusters.mv_gaussian(Float32[0.06259799, 0.15123981, 0.16668582, 248.91315, 861.65173], Float32[0.00033550023 0.00054716866 … -0.36913836 -0.48775765; 0.00054716866 0.0014490675 … -1.439833 -2.5178945; … ; -0.36913836 -1.439833 … 1946.7158 2815.4478; -0.48775765 -2.5178945 … 2815.4478 8824.958], Float32[34590.29 -74969.984 … -6.528216 -2.0861702; -74969.984 332713.56 … 3.773928 1.8949146; … ; -6.528216 3.773928 … 0.0041371784 0.0004609166; -2.0861702 1.8949146 … 0.0004609166 0.00041061162], -14.391136f0, [185.98464352480818 -403.0977038386244 … -0.03510083163123395 -0.011216894896503052; 0.0 412.5842880001881 … -0.025146708261441594 -0.006366189875858899; … ; 0.0 0.0 … 0.03088278507004594 -0.009852609694196254; 0.0 0.0 … 0.0 0.010644951367963566]), DPMMSubClusters.mv_gaussian(Float32[0.1791464, 0.18820712, 0.18291673, 193.54803, 113.60997], Float32[0.0060312203 0.0066885944 … -1.0248829 1.2320757; 0.0066885944 0.0074873995 … -1.1488222 1.3158845; … ; -1.0248829 -1.1488222 … 2789.5645 1856.1819; 1.2320757 1.3158845 … 1856.1819 3922.7844], Float32[40948.582 -65655.49 … 0.2537267 0.44459692; -65655.49 122464.3 … 0.1054923 -1.7483028; … ; 0.2537267 0.1054923 … 0.0006905838 -0.00039544975; 0.44459692 -1.7483028 … -0.00039544975 0.0005345582], -9.6295f0, [202.35756119058647 -324.45287536989406 … 0.0012538533216083275 0.002197085775295539; 0.0 131.1283045712474 … 0.003906926260686424 -0.007896479597351848; … ; 0.0 0.0 … 0.022873913216762718 -0.010823471350514265; 0.0 0.0 … 0.0 0.015966244669007677]), DPMMSubClusters.mv_gaussian(Float32[0.069730125, 0.18153997, 0.20158519, 52.030117, 361.17413], Float32[0.00096561195 0.00091897894 … 0.24272566 0.54286903; 0.00091897894 0.0009927484 … 0.2075526 0.61059624; … ; 0.24272566 0.2075526 … 299.2398 251.77582; 0.54286903 0.61059624 … 251.77582 3077.636], Float32[15775.435 -44926.49 … -0.9968532 0.07147112; -44926.49 223090.77 … -4.760701 0.23708513; … ; -0.9968532 -4.760701 … 0.004564456 -0.00023467139; 0.07147112 0.23708513 … -0.00023467139 0.00038313973], -14.605721f0, [125.60029580432254 -357.69413351544745 … -0.007936710384489865 0.0005690361952558915; 0.0 308.4569229521959 … -0.024637527991528383 0.0014284848339539856; … ; 0.0 0.0 … 0.059906828779382984 -0.004900868949290614; 0.0 0.0 … 0.0 0.018025667980078503]), DPMMSubClusters.mv_gaussian(Float32[0.31085166, 0.34372145, 0.3550167, 136.16328, 397.46524], Float32[0.012530289 0.012478383 … 1.8282605 3.4302857; 0.012478383 0.012857163 … 1.646622 3.1703384; … ; 1.8282605 1.646622 … 3725.9578 3514.3684; 3.4302857 3.1703384 … 3514.3684 6224.187], Float32[3841.826 -9140.254 … -0.583449 0.30646697; -9140.254 36772.18 … 2.81719 -2.1984248; … ; -0.583449 2.81719 … 0.00079870824 -0.00050995394; 0.30646697 -2.1984248 … -0.00050995394 0.0005306287], -6.115148f0, [61.98246557996893 -147.46515278957273 … -0.009413130211812792 0.00494441391883673; 0.0 122.58143240030397 … 0.011658218548910778 -0.011986285861824895; … ; 0.0 0.0 … 0.02396189880505541 -0.013529629180000136; 0.0 0.0 … 0.0 0.012675312775361712]), DPMMSubClusters.mv_gaussian(Float32[0.0344369, 0.16054197, 0.18297413, 54.192654, 893.21643], Float32[0.00018050407 3.5452253e-5 … 0.18343724 0.49512; 3.5452253e-5 0.0003691423 … 0.013330016 -1.3194364; … ; 0.18343724 0.013330016 … 1257.7367 1258.5387; 0.49512 -1.3194364 … 1258.5387 7892.124], Float32[22727.432 -43665.594 … 1.7902502 -4.537464; -43665.594 340222.38 … 0.40942308 1.0202838; … ; 1.7902502 0.40942308 … 0.0016729736 -0.0011619733; -4.537464 1.0202838 … -0.0011619733 0.001836094], -15.715097f0, [150.75620231629006 -289.64376162974213 … 0.011875134307440595 -0.030098025355997805; 0.0 506.28931099888763 … 0.007602336396981622 -0.015203602637267976; … ; 0.0 0.0 … 0.03075772438523142 -0.004904862689218853; 0.0 0.0 … 0.0 0.011256491486710585]), DPMMSubClusters.mv_gaussian(Float32[0.066116154, 0.07037654, 0.069004625, 334.21198, 496.5898], Float32[0.0013134418 0.0013293304 … 0.067262016 -0.020332132; 0.0013293304 0.0016151879 … 0.005262667 0.005077167; … ; 0.067262016 0.005262667 … 1272.0818 -99.24184; -0.020332132 0.005077167 … -99.24184 332.53268], Float32[7314.658 -14404.688 … 0.49670127 0.72564435; -14404.688 45750.824 … -2.6293876 -2.0077295; … ; 0.49670127 -2.6293876 … 0.0010005739 0.0003453907; 0.72564435 -2.0077295 … 0.0003453907 0.0031703354], -12.137415f0, [85.52577603668588 -168.4251074675702 … 0.005807620779847242 0.008484510692477791; 0.0 131.847672857107 … -0.012523835307095057 -0.004389344493301543; … ; 0.0 0.0 … 0.028369920954168938 0.008466785555779031; 0.0 0.0 … 0.0 0.0548381533782948]), DPMMSubClusters.mv_gaussian(Float32[0.31094852, 0.2979832, 0.23720525, 126.71856, 492.6399], Float32[0.0020329466 0.0015396827 … 0.05985472 -0.5377162; 0.0015396827 0.0015745364 … 0.13792315 -0.35614654; … ; 0.05985472 0.13792315 … 364.2714 -125.58155; -0.5377162 -0.35614654 … -125.58155 1018.7845], Float32[2884.0288 -6649.357 … 0.72095156 0.60997456; -6649.357 31802.27 … -1.566036 -0.97441417; … ; 0.72095156 -1.566036 … 0.0031034362 0.00044983218; 0.60997456 -0.97441417 … 0.00044983218 0.0012271743], -11.718041f0, [53.70315389188672 -123.81688043677987 … 0.013424752795899846 0.011358263109818526; 0.0 128.34192405218212 … 0.0007493658016329052 0.0033654672145564603; … ; 0.0 0.0 … 0.053544773568118154 0.006600253423810522; 0.0 0.0 … 0.0 0.031329887792447345]), DPMMSubClusters.mv_gaussian(Float32[0.13280982, 0.23328435, 0.25007954, 172.16208, 935.99976], Float32[0.0011165731 0.0015542377 … -0.09096718 0.24575375; 0.0015542377 0.0022805 … -0.11768176 0.34495455; … ; -0.09096718 -0.11768176 … 1474.727 1338.839; 0.24575375 0.34495455 … 1338.839 3396.7344], Float32[18865.531 -25259.338 … 0.29233062 -0.13670631; -25259.338 142565.8 … -0.11563074 -0.33925933; … ; 0.29233062 -0.11563074 … 0.0010936366 -0.000444404; -0.13670631 -0.33925933 … -0.000444404 0.00048082744], -12.504573f0, [137.35184874222313 -183.90242833712372 … 0.0021283341353594705 -0.0009953000394096396; 0.0 329.7661184059746 … 0.0008362747376633217 -0.0015838419852584058; … ; 0.0 0.0 … 0.032495337946434395 -0.012808192055628703; 0.0 0.0 … 0.0 0.01715810054107613])  …  DPMMSubClusters.mv_gaussian(Float32[0.14750053, 0.2621765, 0.27848405, 247.6828, 475.33728], Float32[0.0043340423 0.006767377 … -0.27033395 3.305858; 0.006767377 0.01227363 … -0.6437674 7.4178934; … ; -0.27033395 -0.6437674 … 667.59186 474.43744; 3.305858 7.4178934 … 474.43744 19022.258], Float32[6804.7803 -21239.549 … -0.54890543 0.17958649; -21239.549 82922.98 … 0.91698194 -0.34275776; … ; -0.54890543 0.91698194 … 0.0018153347 -0.00013185093; 0.17958649 -0.34275776 … -0.00013185093 8.3937426e-5], -6.884441f0, [82.49109179975684 -257.47686854680524 … -0.006654117630670175 0.00217704099381593; 0.0 128.95207460757678 … -0.00617515815231642 0.0016888440048521865; … ; 0.0 0.0 … 0.039050619873314506 -0.0009739682645953894; 0.0 0.0 … 0.0 0.00725051677273192]), DPMMSubClusters.mv_gaussian(Float32[0.1664841, 0.18372695, 0.17506018, 158.29134, 499.86142], Float32[0.0024238804 0.0027158018 … -0.22146866 -0.10770651; 0.0027158018 0.004552405 … -0.52217263 0.2005592; … ; -0.22146866 -0.52217263 … 594.4571 -174.01488; -0.10770651 0.2005592 … -174.01488 1812.6735], Float32[4159.316 -10302.163 … -0.3028964 0.10233165; -10302.163 32349.154 … 0.7389748 0.05888795; … ; -0.3028964 0.7389748 … 0.0019456733 0.00014210319; 0.10233165 0.05888795 … 0.00014210319 0.00058576395], -8.574456f0, [64.49275711542255 -159.7413919568744 … -0.0046965957191057095 0.0015867154523880411; 0.0 82.6549595058374 … -0.0001363008674170931 0.00377898781965155; … ; 0.0 0.0 … 0.041603449009698944 0.00399389035221163; 0.0 0.0 … 0.0 0.02348768517696557]), DPMMSubClusters.mv_gaussian(Float32[0.046129882, 0.12475768, 0.13886578, 59.90002, 31.27576], Float32[0.00070373766 0.0005692013 … -0.07226891 0.08772359; 0.0005692013 0.00061187235 … -0.09737788 0.07829018; … ; -0.07226891 -0.09737788 … 419.40643 -46.119072; 0.08772359 0.07829018 … -46.119072 708.5566], Float32[11944.9 -39572.766 … -0.18496436 -0.017703734; -39572.766 196154.97 … 0.14524397 -0.6952676; … ; -0.18496436 0.14524397 … 0.0024999997 0.00012793801; -0.017703734 -0.6952676 … 0.00012793801 0.0014428355], -15.287576f0, [109.29272908661723 -362.0805010189902 … -0.0016923757119117386 -0.00016198456336599467; 0.0 255.05426462190889 … -0.001833069796385217 -0.0029559163693697334; … ; 0.0 0.0 … 0.04900520853624643 0.0031896886680892363; 0.0 0.0 … 0.0 0.03756753863650139]), DPMMSubClusters.mv_gaussian(Float32[0.32604137, 0.17396684, 0.121498026, 81.9046, 522.5574], Float32[0.010972533 0.006393738 … -0.20583601 -0.10919146; 0.006393738 0.0047903326 … -0.020485569 -0.1336328; … ; -0.20583601 -0.020485569 … 269.91696 -3.9075334; -0.10919146 -0.1336328 … -3.9075334 572.56616], Float32[880.8168 -2642.2444 … 0.30871433 -0.077545054; -2642.2444 10489.3545 … -0.47249383 0.24168873; … ; 0.30871433 -0.47249383 … 0.0038958562 -5.439286e-6; -0.077545054 0.24168873 … -5.439286e-6 0.0017657089], -8.41662f0, [29.678557543096023 -89.02873684414546 … 0.010401931600299178 -0.002612831084525345; 0.0 50.62843489534384 … 0.008958937970012905 0.0001791814830046515; … ; 0.0 0.0 … 0.06087042802554608 0.00041541615259132027; 0.0 0.0 … 0.0 0.041791423299982394]), DPMMSubClusters.mv_gaussian(Float32[0.105200835, 0.22227953, 0.24164462, 75.56501, 533.9345], Float32[0.0016967711 0.0021952828 … -0.520245 0.41093102; 0.0021952828 0.003083224 … -0.7008509 1.4117441; … ; -0.520245 -0.7008509 … 1864.5991 -1656.4943; 0.41093102 1.4117441 … -1656.4943 8153.3643], Float32[22781.742 -58628.6 … 1.4480722 1.2382606; -58628.6 257484.92 … -2.3011918 -0.9064285; … ; 1.4480722 -2.3011918 … 0.0007899476 0.00022893104; 1.2382606 -0.9064285 … 0.00022893104 0.00027289847], -11.147911f0, [150.93621972985696 -388.43295470366087 … 0.009593934803367317 0.008203866626065322; 0.0 326.50385758385727 … 0.004365671378994553 0.0069837571329350295; … ; 0.0 0.0 … 0.025581771808918782 0.005197371055296715; 0.0 0.0 … 0.0 0.011074690018416489]), DPMMSubClusters.mv_gaussian(Float32[0.124847524, 0.21825331, 0.23578511, 72.384285, 706.3923], Float32[0.0011630005 0.0013929514 … -0.54999846 -1.2519826; 0.0013929514 0.001950809 … -0.6639439 -2.0128694; … ; -0.54999846 -0.6639439 … 1236.916 2588.1462; -1.2519826 -2.0128694 … 2588.1462 7448.054], Float32[11112.362 -21089.85 … 4.229297 -1.7022645; -21089.85 144521.61 … -4.46069 0.68511057; … ; 4.229297 -4.46069 … 0.0047077616 -0.0017973; -1.7022645 0.68511057 … -0.0017973 0.00089346885], -12.625441f0, [105.41519111976967 -200.06462336808607 … 0.04012037634179113 -0.016148190546955248; 0.0 323.25802089378135 … 0.011031366597968372 -0.007874734496973733; … ; 0.0 0.0 … 0.0544287956086454 -0.018913622141139185; 0.0 0.0 … 0.0 0.011587202119398746]), DPMMSubClusters.mv_gaussian(Float32[0.08431952, 0.20163491, 0.22212008, 212.60349, 612.6019], Float32[0.0013426365 0.0015058102 … -0.19081156 -1.4855992; 0.0015058102 0.0017895168 … -0.31664184 -1.6345501; … ; -0.19081156 -0.31664184 … 557.7179 113.34354; -1.4855992 -1.6345501 … 113.34354 3771.8367], Float32[23978.965 -55688.86 … -2.0789418 0.7532891; -55688.86 245627.44 … 1.4070058 -0.8206983; … ; -2.0789418 1.4070058 … 0.002362819 7.614644e-5; 0.7532891 -0.8206983 … 7.614644e-5 0.00047676163], -14.134504f0, [154.85143004830275 -359.6276664561771 … -0.013425396368832512 0.0048645923618228825; 0.0 341.02109056070657 … -0.010032042921626872 0.002723420043284989; … ; 0.0 0.0 … 0.0424739731171316 -0.0012763410772641318; 0.0 0.0 … 0.0 0.01628259249216627]), DPMMSubClusters.mv_gaussian(Float32[0.14625637, 0.14667086, 0.13547316, 246.86565, 413.2013], Float32[0.0028411753 0.002844833 … -0.77198654 -1.5246444; 0.002844833 0.002973243 … -0.7013669 -1.2499493; … ; -0.77198654 -0.7013669 … 844.0846 2157.2842; -1.5246444 -1.2499493 … 2157.2842 9004.518], Float32[19401.482 -35265.473 … 0.096237145 0.2633601; -35265.473 79891.37 … 2.105431 -0.51530445; … ; 0.096237145 2.105431 … 0.0039092535 -0.00081006234; 0.2633601 -0.51530445 … -0.00081006234 0.00030056233], -10.708791f0, [139.28920495841004 -253.1816506443503 … 0.0006909160134278007 0.0018907432004206738; 0.0 125.65994720837007 … 0.01814705752156198 -0.00029128586824840293; … ; 0.0 0.0 … 0.05527926906694927 -0.013243696385049497; 0.0 0.0 … 0.0 0.01053828122107371]), DPMMSubClusters.mv_gaussian(Float32[0.10933576, 0.20694555, 0.22053503, 21.896685, 247.87486], Float32[0.0015802934 0.0023545239 … 0.004184123 -0.14794351; 0.0023545239 0.0036051082 … 0.01990607 0.39239636; … ; 0.004184123 0.01990607 … 214.39832 -378.7965; -0.14794351 0.39239636 … -378.7965 22522.05], Float32[36127.863 -55810.35 … 2.9437134 0.631893; -55810.35 209904.34 … -5.4290915 -0.7360766; … ; 2.9437134 -5.4290915 … 0.005057233 0.00013134647; 0.631893 -0.7360766 … 0.00013134647 5.740775e-5], -12.448994f0, [190.07330742560396 -293.62539419625784 … 0.015487253533832446 0.0033244697051460813; 0.0 351.69371518427477 … -0.002506841238835965 0.0006826171527975144; … ; 0.0 0.0 … 0.06933296340322777 0.0011661053940315207; 0.0 0.0 … 0.0 0.006663402213536473]), DPMMSubClusters.mv_gaussian(Float32[0.16336372, 0.2755309, 0.29712486, 250.27747, 874.23303], Float32[0.0025840641 0.003986766 … -0.30248058 0.6464315; 0.003986766 0.00765754 … -0.5349991 1.8576612; … ; -0.30248058 -0.5349991 … 304.51553 -603.0889; 0.6464315 1.8576612 … -603.0889 5179.507], Float32[4669.8843 -15145.96 … 1.8181572 0.04427719; -15145.96 77880.1 … -7.1031213 0.7085091; … ; 1.8181572 -7.1031213 … 0.0053280457 0.00042904832; 0.04427719 0.7085091 … 0.00042904832 0.00028470732], -9.546057f0, [68.33655329169643 -221.6377494519101 … 0.026605925009620036 0.000647928376267666; 0.0 169.57831991262074 … -0.007113196903314074 0.005024902293615391; … ; 0.0 0.0 … 0.06533110935565344 0.007606992552778825; 0.0 0.0 … 0.0 0.013894911713242012])], Float32[0.27812412, 0.0101590995, 0.055434406, 0.06712425, 0.018477904, 0.0051843203, 0.06987033, 0.0075750933, 0.0072897384, 0.039456557  …  0.007776632, 0.0074776057, 0.009470872, 0.00494522, 0.04180987, 0.0346905, 0.032792367, 0.03038033, 0.04390343, 0.004044467], Any[0.07701611518859863, 0.08147692680358887, 0.07693290710449219, 0.07924199104309082, 0.07328391075134277, 0.07474493980407715, 0.07329201698303223, 0.08418893814086914, 0.07021784782409668, 0.07091999053955078  …  0.34993696212768555, 0.3549342155456543, 0.3411440849304199, 0.2701249122619629, 0.35790300369262695, 0.2731451988220215, 0.3360021114349365, 0.36333703994750977, 0.34795594215393066, 0.29085397720336914], Any["no gt", "no gt", "no gt", "no gt", "no gt", "no gt", "no gt", "no gt", "no gt", "no gt"  …  "no gt", "no gt", "no gt", "no gt", "no gt", "no gt", "no gt", "no gt", "no gt", "no gt"], Any[-2.94030621042512e6, -2.94030621042512e6, -2.94030621042512e6, -2.94030621042512e6, -2.94030621042512e6, -2.94030621042512e6, -2.94030621042512e6, -2.94030621042512e6, -2.94030621042512e6, -2.94030621042512e6  …  -1.6481432496118373e6, -1.6480765964690265e6, -1.6474840634485297e6, -1.6474974016742532e6, -1.6480412548153214e6, -1.647426137575488e6, -1.6471668919994659e6, -1.6477608494939622e6, -1.6485810737556033e6, -1.6120752892685952e6], Any[1, 1, 1, 1, 1, 1, 1, 1, 1, 1  …  27, 27, 27, 27, 27, 27, 27, 27, 27, 27])
In [15]:
#Get the cluster color means
color_means = [x.μ[1:3] for x in clusters]

segemnated_image = zeros(3,x,y)
for i=1:x
    for j=1:y
        segemnated_image[:,i,j] = color_means[labels[(i-1)*y+j]]
    end
end
segemnated_image = colorview(RGB,segemnated_image)
Out[15]: