TCL-MAP/
β
βββ data/
β βββ utils.py # λ°μ΄ν° λ‘λ λ° μ μ²λ¦¬μ κ΄λ ¨λ μ νΈλ¦¬ν° ν¨μλ€
β
βββ methods/TCL_MAP/
β βββ SubNets/
β β βββ transformers_encoder/
β β β βββ multihead_attention.py # λ©ν°ν€λ μ΄ν
μ
μ ꡬνν λͺ¨λ
β β β βββ position_embedding.py # ν¬μ§μ
μλ² λ©μ ꡬνν λͺ¨λ
β β β βββ transformer.py # Transformer μΈμ½λ κ΄λ ¨ λͺ¨λ
β β βββ FeatureNets.py # κ° λͺ¨λ¬λ¦¬ν°(ν
μ€νΈ, μ€λμ€, λΉλμ€ λ±)λ‘λΆν° νΌμ²λ₯Ό μΆμΆνλ λ€νΈμν¬λ₯Ό μ μνκΈ° μν λͺ¨λ
β β βββ AlignNets.py # λͺ¨λ¬λ¦¬ν° κ°μ μ λ ¬μ μννλ λ€νΈμν¬λ₯Ό μ μνκΈ° μν λͺ¨λ
β βββ loss.py # μμ€ ν¨μ μ μ (ex. Contrastive Loss)
β βββ manager.py # λͺ¨λΈ νμ΅ κ΄λ¦¬, μ΅ν°λ§μ΄μ λ° μ€μΌμ€λ¬ μ€μ , νλ ¨ λ° νκ° λ‘μ§
β βββ model.py # TCL-MAP λͺ¨λΈμ μ 체 ꡬ쑰 μ μ λ° κ΅¬ν | λͺ¨λλ€μ΄ μ΄λ»κ² μ°κ²°λκ³ λμνλμ§λ₯Ό ꡬνν ν΅μ¬ νμΌ
β
βββ utils/
β βββ functions.py # μ νΈλ¦¬ν° ν¨μλ€ (ER | λͺ¨λΈ μ μ₯, 볡μ, ν
μ€νΈ μ μ₯)
β βββ metrics.py # νκ° λ©νΈλ¦μ€ μ μ (μ νλ, F1 μ€μ½μ΄ λ±)
Original source: https://github.com/thuiar/TCL-MAP