117 lines
9.2 KiB
Plaintext
117 lines
9.2 KiB
Plaintext
|
a
|
|||
|
|
EZ>jB. ã @ sx d Z ddlmZmZmZmZ ddlmZ ddlZeG dd„ dƒƒZ G dd„ dƒZ
|
|||
|
|
G d d
|
|||
|
|
„ d
|
|||
|
|
ƒZ d d „ Z e
|
|||
|
|
d
|
|||
|
|
krte ƒ dS )zP
|
|||
|
|
Reality DNA
|
|||
|
|
Unique fingerprint for each place
|
|||
|
|
Enables global similarity search
|
|||
|
|
é )ÚDictÚListÚOptionalÚTuple)Ú dataclassNc @ s¤ e Zd ZU dZeed< eed< eeef ed< eed< eed< eed< eed< eed < eed
|
|||
|
|
< eed < eed < eed
|
|||
|
|
< eed< ee ed< eed< edœdd„Z dS )Ú
|
|||
|
|
RealityDNAzDNA fingerprint of a placeÚ location_idÚ
|
|||
|
|
location_nameÚ coordinatesÚ walkabilityÚtourismÚretailÚfamilyÚnoiseÚhistoryÚsafetyÚgreeneryÚ connectivityÚ
|
|||
|
|
affordabilityÚ
|
|||
|
|
dna_vectorÚdna_hash©Úreturnc C sŒ | j | j| jt| jdƒt| jdƒt| jdƒt| jdƒt| jdƒt| j dƒt| j
|
|||
|
|
dƒt| j dƒt| j dƒt| j
|
|||
|
|
dƒdœ
|
|||
|
|
dd„ | jD ƒ| jdœS )Né )
|
|||
|
|
r r r
|
|||
|
|
r r r r r r r c S s g | ]}t |d ƒ‘qS )é )Úround©Ú.0Úv© r ú?/home/bernt/.openclaw/workspace/iom/intelligence/reality_dna.pyÚ
|
|||
|
|
<listcomp>4 ó z&RealityDNA.to_dict.<locals>.<listcomp>)r r r
|
|||
|
|
Údnar r )r r r
|
|||
|
|
r r r r
|
|||
|
|
r r r r r r r r r ©Úselfr r r Úto_dict# s"
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
ö ïzRealityDNA.to_dictN)
|
|||
|
|
Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__Ú__doc__ÚstrÚ__annotations__r Úfloatr r& r r r r r s"
|
|||
|
|
r c @ s@ e Zd ZdZeeeeef edœdd„Ze e edœdd„Z
|
|||
|
|
dS ) ÚRealityDNABuilderz Builds Reality DNA from signals)r r Úsignalsr c C sÒ | dd¡}| dd¡}| dd¡}| dd¡}| dd¡}| dd¡} | dd¡}
|
|||
|
|
| d d¡} | d
|
|||
|
|
d¡} d | d d¡ }
|
|||
|
|
|||||| |
|
|||
|
|
| | |
|
|||
|
|
g
|
|||
|
|
}| |¡}t||| d
|
|||
|
|
dddœ¡|||||| |
|
|||
|
|
| | |
|
|||
|
|
||d�S )z Build DNA from signal valuesr é2 r Úretail_densityÚfamily_friendlyÚ noise_levelÚhistorical_valueÚ safety_indexr Úmobile_connectivityéd Ú rent_burdenr
|
|||
|
|
r ©ÚlatÚlng)r r r
|
|||
|
|
r r r
|
|||
|
|
r r r r r r r r r )ÚgetÚ _hash_vectorr )r% r r r/ r r r
|
|||
|
|
r r r r r r r Úvectorr r r r Úbuild< s>
|
|||
|
|
|
|||
|
|
þ
|
|||
|
|
ñzRealityDNABuilder.build)r> r c C s" dd„ |D ƒ}d dd„ |D ƒ¡S )zCreate hash from DNA vectorc S s g | ]}t |d ƒ‘qS )é
|
|||
|
|
)Úintr r r r r! i r" z2RealityDNABuilder._hash_vector.<locals>.<listcomp>Ú c s s | ]}t t|d ƒƒV qdS )é N)r+ Úmin)r Úqr r r Ú <genexpr>j r" z1RealityDNABuilder._hash_vector.<locals>.<genexpr>)Újoin)r% r> Z quantizedr r r r= f s z RealityDNABuilder._hash_vectorN) r' r( r) r* r+ r r- r r? r r= r r r r r. 9 s *r. c @ s® e Zd ZdZdd„ Zedœdd„Zd eeee e
|
|||
|
|
d œd
|
|||
|
|
d „Z d e
|
|||
|
|
e e
|
|||
|
|
f ee e
|
|||
|
|
d œd
|
|||
|
|
d„Zd eee e
|
|||
|
|
dœdd„Ze e
|
|||
|
|
e e
|
|||
|
|
e
|
|||
|
|
dœdd„Ze
|
|||
|
|
dœdd„Ze
|
|||
|
|
dœdd„ZdS ) ÚGlobalSimilaritySearchz Find similar places globallyc C s
|
|||
|
|
i | _ d S ©N)Ú dna_databaser$ r r r Ú__init__p s z GlobalSimilaritySearch.__init__)r# c C s || j |j< dS )zAdd a place to the databaseN)rJ r )r% r# r r r Ú add_places s z GlobalSimilaritySearch.add_placeé T)Ú query_dnaÚtop_kÚ exclude_selfr c C s€ g }| j ¡ D ]R\}}|r&||jkr&q| |j|j¡}| |j|j|jt|dƒ| ¡ d dœ¡ q|j
|
|||
|
|
dd„ dd� |d|… S ) z‚
|
|||
|
|
Find places similar to query
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
List of similar places with similarity scores
|
|||
|
|
é r# ©r r r
|
|||
|
|
Ú
|
|||
|
|
similarityr# c S s | d S ©NrS r ©Úxr r r Ú<lambda>• r" z5GlobalSimilaritySearch.find_similar.<locals>.<lambda>T©ÚkeyÚreverseN© rJ Úitemsr Ú_calculate_similarityr Úappendr r
|
|||
|
|
r r& Úsort)r% rN rO rP Ú similaritiesr Ú candidaterS r r r Ú find_similarw s
|
|||
|
|
û
|
|||
|
|
z#GlobalSimilaritySearch.find_similar)ÚpatternrO r c
|
|||
|
|
C sÖ | dd¡| dd¡| dd¡| dd¡| dd¡| dd¡| dd¡| d d¡| d
|
|||
|
|
d¡| d d¡g
|
|||
|
|
}g }| j ¡ D ]@\}}| ||j¡}| |j|j|jt |d ƒ|
|
|||
|
|
¡ d
|
|||
|
|
dœ¡ qv|j dd„ dd� |d|… S )u±
|
|||
|
|
Find places matching a DNA pattern
|
|||
|
|
|
|||
|
|
Example:
|
|||
|
|
{"walkability": 90, "tourism": 80, "history": 95}
|
|||
|
|
→ Finds places like Gamla Stan
|
|||
|
|
r r0 r r
|
|||
|
|
r r r r r r r rQ r# rR c S s | d S rT r rU r r r rW À r" z<GlobalSimilaritySearch.find_by_dna_pattern.<locals>.<lambda>TrX N) r< rJ r\ r] r r^ r r r
|
|||
|
|
r r& r_ )r% rc rO Z query_vectorr` r ra rS r r r Úfind_by_dna_pattern™ s.
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
ö
|
|||
|
|
|
|||
|
|
û
|
|||
|
|
z*GlobalSimilaritySearch.find_by_dna_patternrQ )rN rO r c C sz g }| j ¡ D ]N\}}||jkr"q| |j|j¡}| |j|j|jt|dƒ| ¡ d dœ¡ q|j
|
|||
|
|
dd„ d� |d|… S )z&Find places that are opposite to queryrQ r# rR c S s | d S rT r rU r r r rW Û r" z7GlobalSimilaritySearch.find_contrasts.<locals>.<lambda>)rY Nr[ )r% rN rO r` r ra rS r r r Úfind_contrastsÄ s
|
|||
|
|
|
|||
|
|
û
|
|||
|
|
z%GlobalSimilaritySearch.find_contrasts)Úvector_aÚvector_br c C sx t |ƒt |ƒkrdS tdd„ t||ƒD ƒƒ}tdd„ |D ƒƒd }tdd„ |D ƒƒd }|dksh|dkrldS ||| S )z3Calculate cosine similarity between two DNA vectorsg c s s | ]\}}|| V qd S rI r )r ÚaÚbr r r rF å r" z?GlobalSimilaritySearch._calculate_similarity.<locals>.<genexpr>c s s | ]}|| V qd S rI r )r rh r r r rF æ r" g à?c s s | ]}|| V qd S rI r )r ri r r r rF ç r" r )ÚlenÚsumÚzip)r% rf rg Z dot_productZ magnitude_aZ magnitude_br r r r] ß s z,GlobalSimilaritySearch._calculate_similarityr c C s t | jƒ| ¡ dœS )zGet database statistics)Z total_placesZcoverage)rj rJ Ú_calculate_coverager$ r r r Ú get_statsî s þz GlobalSimilaritySearch.get_statsc C st | j s
|
|||
|
|
i S dd„ | j ¡ D ƒ}dd„ | j ¡ D ƒ}t|ƒt|ƒgt|ƒt|ƒgt|ƒt|ƒ t|ƒt|ƒ dœdœS )z Calculate geographic coveragec S s g | ]}|j d ‘qS )r: ©r
|
|||
|
|
©r Údr r r r! ú r" z>GlobalSimilaritySearch._calculate_coverage.<locals>.<listcomp>c S s g | ]}|j d ‘qS )r; ro rp r r r r! û r" r9 )Z lat_rangeZ lng_rangeÚcenter)rJ ÚvaluesrD Úmaxrk rj )r% ZlatsZlngsr r r rm õ s þýz*GlobalSimilaritySearch._calculate_coverageN)rM T)rM )rQ )r' r( r) r* rK r rL rA Úboolr r rb r+ r- rd re r] rn rm r r r r rH m s4 üû % ý
|
|||
|
|
ü . ýü rH c
|
|||
|
|
C sø t ƒ } tƒ }ddddddddd d
|
|||
|
|
d d d
|
|||
|
|
œdœ }| dd|¡}| |¡ tdƒ td|j› �ƒ td|j› �ƒ ddddd dddd dddd d d
|