Karate Club: An API Oriented Open-source Python Framework for Unsupervised Learning on Graphs
arXiv:2003.04819
Abstract
We present Karate Club a Python framework combining more than 30 state-of-the-art graph mining algorithms which can solve unsupervised machine learning tasks. The primary goal of the package is to make community detection, node and whole graph embedding available to a wide audience of machine learning researchers and practitioners. We designed Karate Club with an emphasis on a consistent application interface, scalability, ease of use, sensible out of the box model behaviour, standardized dataset ingestion, and output generation. This paper discusses the design principles behind this framework with practical examples. We show Karate Club's efficiency with respect to learning performance on a wide range of real world clustering problems, classification tasks and support evidence with regards to its competitive speed.
arXiv commentThe frameworks is available at: https://github.com/benedekrozemberczki/karateclub
References in corpus (12)
- Scikit-learn: Machine Learning in Python
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- The NumPy array: a structure for efficient numerical computation
- Cooperative Game Theory Approaches for Network Partitioning
- Near linear time algorithm to detect community structures in large-scale networks
- API design for machine learning software: experiences from the scikit-learn project
- graph2vec: Learning Distributed Representations of Graphs
- Learning Structural Node Embeddings Via Diffusion Wavelets
- NetLSD: Hearing the Shape of a Graph
- EdMot: An Edge Enhancement Approach for Motif-aware Community Detection
- A Simple Baseline Algorithm for Graph Classification
- FSCNMF: Fusing Structure and Content via Non-negative Matrix Factorization for Embedding Information Networks