a ÁdI ã@sžddlZddlmZddlZdd„Zdgdeƒfdd„Zdd„Zd d „Z d d „Z e d krše  gd¢gd¢gd¢gd¢gd¢g¡Z gd¢Zee e eƒdƒdS)éNcCs2|dd…|f}t |¡}||dkr*|SdSdS)Nr)ÚnpÚargmax)ÚmatrixÚnodeÚparentsZ max_parent©rú5/mnt/chromeos/MyFiles/work/nlp_tsa/tsa/TSA/outline.pyÚ find_parents   r éc Cstg}t||ƒ}|durp||vrp| ||g¡|D]}| ||g¡q0||krp| |¡| t|||d|d¡|S)Nr )Úvisited)r ÚappendÚaddÚextendÚ find_tree) rrÚdepthÚchildrenÚ max_depthr ÚresultÚparentÚchildrrrrs  rcCs&d}|D]\}}||||9}q|S©Nr r)ÚtreerÚprobrrrrrÚ find_probs rcCsi}tt|ƒƒD]”}||}g}tt|ƒƒD]}||dkr,| |¡q,t|||d}tdd„|Dƒƒ}|rt||ƒ}t|ƒ|vr˜|t|ƒ|7<q||t|ƒ<qt| ¡dd„dd} | d\} }i} | D]*\} } | | vrð| |  | ¡qÐ| g| | <qÐ| |fS) Nr)rcSsg|] }t|ƒ‘qSr)Útuple)Ú.0ÚxrrrÚ +óz find_forests..cSs|dSrr)rrrrÚ2rzfind_forests..T)ÚkeyÚreverse)ÚrangeÚlenr rrrÚsortedÚitems)rÚkZforestsÚirZ child_listÚjrrZsorted_forestsZforestrrrrrrÚ find_forests"s,       r)c s¤t|dƒ\}}t||ƒi}| ¡D]}‡fdd„||Dƒ||<q$g}g}t| ¡ƒD]B}| d¡| d¡||D] } | | ¡| d| ›d¡qxqX||fS)Nr csg|] }ˆ|‘qSrr)rr'©Ú sentencesrrrCrz#passage_outline..u 主题:u 主题: z- Ú )r)ÚprintÚkeysr$r ) rr+rrÚ structureÚeachÚoutlÚ outline_listr Úsentencerr*rÚpassage_outline=s      r4Ú__main__)çgÞ´2D•?gˆO×f¥A»?g€œQoN¸?gsÉ‚Ó̤±?)g©qˆd?r6gC’aO®?gŸ“Þ7¾ö ?gžTEWÃ?)g¸H éŽ?g ±ò²Ãë?r6gi¦ÿÕx#¨?g&†‰«o§?)g ®$?f‘?g6:ç§8 ?gt}¢¤?r6gÏ¡ U1•Ž?)g¤«M⤗?gá/ÁÉ–œ?gÖÝÀ ÆÂ°?gz–9À@ε?r6)u 主题å¥å­1u 主题å¥å­2u 主题å¥å­3u 主题å¥å­4u 主题å¥å­5)ÚnumpyrÚmatplotlib.pyplotÚpyplotÚpltr Úsetrrr)r4Ú__name__Úarrayrr+r-rrrrÚs    ü