Stars
⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
A scikit-learn based module for multi-label et. al. classification
A Python nearest neighbor descent for approximate nearest neighbors
Creating beautiful plots of data maps
Pruna is a model optimization framework built for developers, enabling you to deliver faster, more efficient models with minimal overhead.
[CONTRIBUTORS WELCOME] Generalized Additive Models in Python
SSSegmentation: An Open Source Supervised Semantic Segmentation Toolbox Based on PyTorch.
PaCMAP: Large-scale Dimension Reduction Technique Preserving Both Global and Local Structure
KDD 2019: Robust Anomaly Detection for Multivariate Time Series through Stochastic Recurrent Neural Network
A paper summary of image inpainting
The Tensorflow, Keras implementation of U-net, V-net, U-net++, UNET 3+, Attention U-net, R2U-net, ResUnet-a, U^2-Net, TransUNET, and Swin-UNET with optional ImageNet-trained backbones.
Implementation of the Leiden algorithm for various quality functions to be used with igraph in Python.
Configurable Generation of Synthetic Schemas and Knowledge Graphs at Your Fingertips
Code release for CVPR'24 submission 'OmniGlue'
Shapley Interactions and Shapley Values for Machine Learning
A library for generating and evaluating synthetic tabular data for privacy, fairness and data augmentation.
A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.
Nimfa: Nonnegative matrix factorization in Python
Self-supervised Equivariant Attention Mechanism for Weakly Supervised Semantic Segmentation, CVPR 2020 (Oral)
Fast SHAP value computation for interpreting tree-based models
Optimal binning: monotonic binning with constraints. Support batch & stream optimal binning. Scorecard modelling and counterfactual explanations.