sample.py

import numpy as np
import networkx as nx
from scipy import sparse
from label_propagation import LGC,HMN,PARW,OMNI,CAMLP
from sklearn.model_selection import GridSearchCV
from sklearn.model_selection import train_test_split

G = nx.karate_club_graph()
labels = {'Officer':0, 'Mr. Hi':1}
nodes = np.array([(n,labels[attr['club']]) for n,attr in G.nodes(data=True)])
x = nodes[:,0]
y = nodes[:,1]
A = nx.to_scipy_sparse_array(G,nodelist=x)

x_train, x_test, y_train, y_test = train_test_split(x,y,test_size=0.5,random_state=0)

methods = [
    ('HMN', HMN(), {'graph':[A]}),
    ('LGC', LGC(), {'graph':[A], 'alpha':[0.01,0.05,0.1,0.5,0.99]}),
    ('PARW', PARW(), {'graph':[A], 'lamb':[0.01, 0.05, 0.01, 0.5, 0.99]}),
    ('OMNIProp', OMNI(), {'graph':[A], 'lamb':[0.01, 0.1, 1.0, 10.0, 100.0]}),
    ('CAMLP', CAMLP(), {'graph':[A], 'beta':[0.01, 0.1, 1.0, 10.0, 100.0], 'H':[np.array([[1,0],[0,1]]), np.array([[0,1],[1,0]])]})
]

for name, clf, params in methods:
    print("=========================")
    print(name)
    gs = GridSearchCV(clf, params, cv=5)
    gs.fit(x_train,y_train)

    print()
    print("Grid Scores:")
    for score in gs.cv_results_:
        print( score)

    model = gs.best_estimator_
    model.fit(x_train,y_train)
    print("\nBest Estimator:")
    print(model)

    predicted = model.predict(x_test)
    acc = (predicted==y_test).mean()
    print("\nAccuracy: %s" % acc)
    print()