՟Kc @sddkZddkZddkZddkZddklZy+ddklZlZl Z dGHe Z WndGHe Z nXdddYZ dS( iN(tadd(t cPropagatet cEuclideant cUpdateUnitsloading SOM C-extensionscan not load SOM C-extensiontSOMcBsUeZdddddddZddd Zd Zdddd Zd Zd ZddZdZ ddZ dZ dZ dZ dZdZdZdZdZdZddhdZddedhdZd Zd!d"Zd#d!dd!d$Zd%Zded&Zed'Zd(d)ZRS(*iiithexttoroidtcCsG||_||_|djo|i|n|i|||dS(NR(tgridTypet topologyTypetreadSOMtresize(tselftinputDimtnumRowstnumColstgridttopologyt initialFile((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pyt__init__s    tArrayicCsg}|djo |i}n|djoOh}xAt|iD]0}x't|iD]}g|||ft|iD]-}d|i|i||||!d GqWHHqWnGxC|iD]8\}}|G|Gd G|i|i||||!GHq!WdS( NsRows = %d, Cols = %dsgrid = %s inputDim = %diiRt is [s] s ( RRRR tlent prettyFloatR$RR%(R t dimensiontminDimtmaxDimtspacerRR((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pytdescribes*  /+  cCsMd}|i||}x/t||D]\}}|||d7}q'W|S(Ngi(R$tzip(R t inputVectortoutRowtoutColt distSquaredt currentUnittinputtcurrent((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pytprops  cCsEd}x/t||D]\}}|||d7}qWti|S(Ngi(RLtmathtsqrt(R tvecAtvecBtdistancetatb((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pyt euclideans  c Csd}|i}d}x|D]x\}}d}|i||}x/t||D]\} } || | d7}qOW||jo||f}|}qqW|S(Niii(ii(R%R$RL( R t inputVecttsmallesttlocUnitstwinUnitRRt magSquaredRQRRRS((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pyt propagates     c Csut||}t||}|i|}||jo |}n|i|}||jo |}nt||S(N(tabsRRtmax( R taRowtaColtbRowtbColtdistRowtdistColtdiffRowtdiffCol((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pyt recDistances      c Cs|d}|d}|d}|d}g}t}d|jot}|i||fnt} ||ijot} |i||fnt} d|jot} |i||fnt} ||ijot} |i||fn|ddjoK| o@|o|i||fn| o|i||fqLqnH| o@|o|i||fn| o|i||fqn|S(Niii(tFalsetTrueRRR( R tarowtacoltarowmtarowptacolmtacolptresultstarowmOKtarowpOKtacolmOKtacolpOK((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pytimmediateHexNeighborssB      cCs|d|i}|d|i}|d|i}|d|i}||f||f||f||fg}|ddjo*|i||f|i||fn'|i||f|i||f|S(Niii(RRR(R RpRqRrRsRtRuRv((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pytimmediateHexNeighborsToroids* cCsg}|||fd}||fg|||fd}td|d}x|D]} |igg} g} x|D]\} } xy|| | fdD]c\} } | | f|joD|di| | f|id| | f| i| | fqqWq{W| }qUW|S(Niiii(RRtinsert(R ttheRowttheColt startDictt maxRadiust resultListt previousUnitst seenUnitst radiusListtaRadiust currentUnitst nextUnitsRR((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pytgetHexNeighbors s&!    c CsZ|idd}t|d}|idjox!|iD]\}}g}|D] }|gqS~|||f      +   cCs|idd}toHxt|D]9}|t|i|||i|i|ii|q Wn0x,|D]$}||i||i|qhW|S(NRR( R!RRR$RRR RRb(R RRtrecID((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pyt getWinnerss;"c Csd}|i}g}h}d}t} |i} to t} n|djot} t|d} n|i|}x|iD]} | \} }|i | |}|o"dt | |i |fGHn| o-| i d| ||i |ddfnx|| D]}| |||}||7}|o#d ||i |||fGHn| o1| i d |||i ||ddfq q WqW| o| i ny|t|}Wn d}nX|S( NigRR5s%s: [%s]sunit %d,%d %s tdelims s %s [%s] %.2fs %.2f %s %s (R RnR\RRRoR(RR%R$RRFR6R-RE(R tdatasett outfilenameRRRGtidListt winnerDictt totalScoret doOutfilet scorefuncR7R RRt unitWeightsRt winnerScorettoreturn((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pyRsH      "-  #9 c Cs|idd}t|}x|D]}|i}d|joE|did\}}t|}t|}g|||fRCRKRTR\RbRmR{R|RRRRRnRRRRRRRRF(((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pyRs6         *      # + ((RURRtcopytoperatorRt_csomRRRRoRRnR(((s8/woldlab/castor/data00/home/georgi/commoncode/SOM/som.pyts0