概述

Scikit-network 是一个用于图上机器学习的开源 Python 包。

每个图都由其在 scipy 的稀疏 CSR 格式中的邻接矩阵表示。

notebook 介绍了该软件包的概述。

安装

要安装 scikit-network,请在您的终端中运行以下命令:

$ pip install scikit-network

如果您还没有安装 pip,请参考 Python 安装指南

或者,您可以从 Github 下载源代码并运行:

$ cd <scikit-network folder>
$ python setup.py develop

导入

在 Python 中导入 scikit-network:

import sknetwork as skn

使用

图由其 邻接 矩阵(方阵)表示。当图是二部的时,它可以用其 二部邻接 矩阵(矩形矩阵)表示。请查看我们的 教程,了解加载图的各种方法(例如,从边列表、数据框或 CSV 文件加载)。

每个算法都用一个带有 fit 方法的对象表示。

以下是一个使用 Louvain 算法空手道俱乐部图 进行聚类的示例:

from sknetwork.data import karate_club
from sknetwork.clustering import Louvain

adjacency = karate_club()
algorithm = Louvain()
algorithm.fit(adjacency)

教程 提供了更多详细信息。