17c1ac88b1
Complete the reproduction of the Raw-feature-extractor: The purpose of read_idaFILE.py is to read the raw-feature from the generated .ida file and display
27 lines
619 B
Python
27 lines
619 B
Python
# -*- coding: UTF-8 -*-
|
||
import networkx as nx
|
||
import pdb
|
||
def betweeness(g):
|
||
#pdb.set_trace()
|
||
betweenness = nx.betweenness_centrality(g)
|
||
#print betweenness
|
||
return betweenness #list
|
||
|
||
def eigenvector(g):
|
||
centrality = nx.eigenvector_centrality(g)
|
||
return centrality
|
||
|
||
def closeness_centrality(g):
|
||
closeness = nx.closeness_centrality(g)
|
||
return closeness
|
||
|
||
def retrieveGP(g): #list，转化为float。将基本块级别的betweeness转化为函数级别的betweeness
|
||
bf = betweeness(g)
|
||
#close = closeness_centrality(g)
|
||
#bf_sim =
|
||
#close_sim =
|
||
x = sorted(bf.values())
|
||
value = sum(x)/len(x)
|
||
return round(value,5)
|
||
|