InfluenceDiffusion 0.0.7__tar.gz → 0.0.8__tar.gz

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/InfluenceDiffusion/Graph.py

@@ -3,6 +3,8 @@ import numpy as np
 from typing import List, Tuple, Union, Dict
 import networkx as nx
 
+__all__ = ["Graph"]
+
 
 class Graph(nx.DiGraph):
     def __init__(self, edge_list: List[Tuple[int, int]],

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/InfluenceDiffusion/Trace.py

@@ -6,6 +6,8 @@ from itertools import chain
 from .Graph import Graph
 from .utils import multiple_union
 
+__all__ = ["PseudoTraces", "Traces", "Trace"]
+
 
 class PseudoTraces(dict):
     """A collection of pseudo traces.

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/InfluenceDiffusion/__init__.py

@@ -1,4 +1,4 @@
 from .Graph import Graph
 from .influence_models import InfluenceModel, ICM, LTM, GLTM
 from .Trace import Trace, Traces, PseudoTraces
-from .utils import invert_non_zeros, multiple_union
+from .utils import invert_non_zeros, multiple_union, random_vector_inside_simplex, random_vector_on_simplex

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/InfluenceDiffusion/estimation_models/BaseWeightEstimator.py

@@ -11,6 +11,8 @@ from ..Trace import Traces, PseudoTraces
 from ..Graph import Graph
 from ..utils import multiple_union
 
+__all__ = ["BaseWeightEstimator", "BaseGLTWEightEstimator", "BaseICWEightEstimator"]
+
 
 class BaseWeightEstimator:
     """Base class for weight estimation in influence models.

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/InfluenceDiffusion/estimation_models/EMEstimation.py

@@ -6,6 +6,8 @@ from ..Trace import Traces, PseudoTraces
 from ..utils import invert_non_zeros
 from ..Graph import Graph
 
+__all__ = ["BaseWeightEstimatorEM", "ICWeightEstimatorEM", "LTWeightEstimatorEM"]
+
 
 class BaseWeightEstimatorEM(BaseWeightEstimator):
     """Base class for EM weight estimators.

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/InfluenceDiffusion/estimation_models/OptimEstimation.py

@@ -7,6 +7,8 @@ from ..Graph import Graph
 from ..Trace import Traces, PseudoTraces
 from .BaseWeightEstimator import BaseGLTWEightEstimator
 
+__all__ = ["GLTGridSearchEstimator", "GLTWeightEstimator"]
+
 
 class GLTWeightEstimator(BaseGLTWEightEstimator):
     """GLTW Weight Estimator for modeling influence in networks.

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/InfluenceDiffusion/utils.py

@@ -14,3 +14,17 @@ def invert_non_zeros(array):
     non_zero_mask = array != 0
     out[non_zero_mask] = 1. / out[non_zero_mask]
     return out
+
+
+def random_vector_inside_simplex(dim, ub=1):
+    U = np.random.uniform(low=0, high=ub, size=dim)
+    U_sorted = np.sort(U)
+    U_sorted = np.concatenate(([0], U_sorted))
+    x = np.diff(U_sorted)
+    return x
+
+
+def random_vector_on_simplex(dim, ub=1):
+    X = random_vector_inside_simplex(dim=dim, ub=1)
+    return X / np.sum(X) * ub
+

InfluenceDiffusion-0.0.8/InfluenceDiffusion/weight_samplers.py

@@ -0,0 +1,30 @@
+import numpy as np
+
+from .Graph import Graph
+from .utils import random_vector_inside_simplex, random_vector_on_simplex
+
+
+def make_weighted_cascade_weights(g: Graph):
+    vertex_2_indegree = g.get_vertex_2_indegree_dict(weighted=False)
+    weights = 1. / np.array([vertex_2_indegree[sink] for sink in g.edge_array[:, 1]])
+    return weights
+
+
+def make_random_weights_with_indeg_constraint(g: Graph, indeg_ub=1, random_state=None):
+    if random_state:
+        np.random.seed(random_state)
+    weights = np.zeros(g.count_edges())
+    for v in g.get_sinks():
+        parent_mask = g.get_parents_mask(v)
+        weights[parent_mask] = random_vector_inside_simplex(parent_mask.sum(), ub=indeg_ub)
+    return weights
+
+
+def make_random_weights_with_fixed_indeg(g: Graph, indeg_ub=1, random_state=None):
+    if random_state:
+        np.random.seed(random_state)
+    weights = np.zeros(g.count_edges())
+    for v in g.get_sinks():
+        parent_mask = g.get_parents_mask(v)
+        weights[parent_mask] = random_vector_on_simplex(parent_mask.sum(), ub=indeg_ub)
+    return weights

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/InfluenceDiffusion.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: InfluenceDiffusion
-Version: 0.0.7
+Version: 0.0.8
 Summary: InfluenceDiffusion package
 Author: Alexander Kagan
 Author-email: <amkagan@umich.edu>

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/InfluenceDiffusion.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ InfluenceDiffusion/Trace.py
 InfluenceDiffusion/__init__.py
 InfluenceDiffusion/influence_models.py
 InfluenceDiffusion/utils.py
+InfluenceDiffusion/weight_samplers.py
 InfluenceDiffusion.egg-info/PKG-INFO
 InfluenceDiffusion.egg-info/SOURCES.txt
 InfluenceDiffusion.egg-info/dependency_links.txt

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: InfluenceDiffusion
-Version: 0.0.7
+Version: 0.0.8
 Summary: InfluenceDiffusion package
 Author: Alexander Kagan
 Author-email: <amkagan@umich.edu>

{InfluenceDiffusion-0.0.7 → InfluenceDiffusion-0.0.8}/setup.py

@@ -1,6 +1,6 @@
 from setuptools import setup, find_packages
 
-VERSION = '0.0.7'
+VERSION = '0.0.8'
 DESCRIPTION = 'InfluenceDiffusion package'
 LONG_DESCRIPTION = 'in this package, we implement popular network diffusion models and methods for their estimation.'