def test__liteqclass(self): u = 1 set_of_u_1 = zkl.load("{}/liteqclass_output_n_5_u_1.zkl".format(self._ABS_PATH)) g2 = bfutils.undersample(self._G, u) s = ur.liteqclass(g2, verbose=False, capsize=1000) self.assertEqual(s, set_of_u_1) u = 2 set_of_u_2 = zkl.load("{}/liteqclass_output_n_5_u_2.zkl".format(self._ABS_PATH)) g2 = bfutils.undersample(self._G, u) s = ur.liteqclass(g2, verbose=False, capsize=1000) self.assertEqual(s, set_of_u_2)
from gunfolds.tools.bfutils import undersample from gunfolds.tools import ecj from gunfolds.tools.testgraphs import * from gunfolds.tools import zickle import igraph import numpy as np import os import scipy import StringIO DIR_NAME = os.path.dirname(__file__) ABS_PATH = os.path.abspath(os.path.join(DIR_NAME)) colors = zickle.load('{}/../data/colors.zkl'.format(ABS_PATH)) def graph2dict(g): D = {}` for v in range(0, len(g.vs)): D[g.vs[v]["label"]] = {} for u in g.neighbors(v, mode="OUT"): D[g.vs[v]["label"]][g.vs[u]["label"]] = set([(0, 1)]) return D def paintSCC(g, cm): nameidx = {} D = graph2dict(g) scc = ecj.scc(D) for i in range(0, len(scc)): for v in scc[i]: g.vs[g.vs['label'].index(v)]["color"] = cm[i]
for k in keys: c[k] = len(np.where(eqc == k)[0]) return c if __name__ == '__main__': import sys sys.path.append('/na/homes/splis/soft/src/dev/tools/stackedBarGraph/') from stackedBarGraph import StackedBarGrapher SBG = StackedBarGrapher() fig = pl.figure(figsize=[10, 1.3]) # Read in data & create total column d = zkl.load("hooke_nodes_6_g32g1_.zkl") # hooke_nodes_35_newp_.zkl") densities = np.sort(d.keys()) # unique size usz = set() dc = {} for u in densities: dc[u] = get_counts(d[u]) for v in dc[u]: usz.add(v) for u in densities: for c in usz: if not c in dc[u]: dc[u][c] = 0
def listplot(fname, mname='JJ', stl='', width=5): l = zkl.load(fname) y = min(width, len(l)) x = np.int(np.ceil(len(l) / float(y))) d2l.matrix_list(l, y, x, R=2, w_gap=1, h_gap=2, mname=mname, stl=stl)
def timesfromfile(fname): d = zkl.load(fname) return getalltimes(d)
def loadgraphs(fname): g = zkl.load(fname) return g
def gettimes(d): t = [x['ms'] for x in d] time = map(lambda x: x / 1000. / 60., t) return time l = ['leibnitz_nodes_15_density_0.1_newp_.zkl', 'leibnitz_nodes_20_density_0.1_newp_.zkl', 'leibnitz_nodes_25_density_0.1_newp_.zkl', 'leibnitz_nodes_30_density_0.1_newp_.zkl', 'leibnitz_nodes_35_density_0.1_newp_.zkl'] if __name__ == '__main__': alltimes_new = [] for fname in l: d = zkl.load(fname) alltimes_new.append(gettimes(d)) shift = 0.15 wds = 0.3 fliersz = 2 lwd = 1 plt.figure(figsize=[10, 2]) g = sb.boxplot(alltimes_new, names=map(lambda x: str(int(x * 100)) + "", densities), widths=wds, color="Reds", fliersize=fliersz, linewidth=lwd, **{'positions': np.arange(len(densities)) + shift, 'label': 'MSL'})
import sys import os sys.path.append(os.path.expanduser(SBDIR)) from stackedBarGraph import StackedBarGrapher SBG = StackedBarGrapher() l = [(0.15, 'leibnitz_nodes_15_density_0.1_newp_.zkl'), (0.20, 'leibnitz_nodes_20_density_0.1_newp_.zkl'), (0.25, 'leibnitz_nodes_25_density_0.1_newp_.zkl'), (0.30, 'leibnitz_nodes_30_density_0.1_newp_.zkl'), (0.35, 'leibnitz_nodes_35_density_0.1_newp_.zkl')] fig = pl.figure(figsize=[10, 3]) # Read in data & create total column d = zkl.load("hooke_nodes_6_g32g1_.zkl") # hooke_nodes_35_newp_.zkl") densities = [.15, .20, .25, .30, .35] d = {} for fname in l: d[fname[0]] = zkl.load(fname[1]) # unique size usz = set() dc = {} for u in densities: dc[u] = get_counts(d[u]) for v in dc[u]: usz.add(v) for u in densities: for c in usz: