jianzhu Goto Github PK

a-pgnn

Source code and datasets for the paper "Personalizing Graph Neural Networks with Attention Mechanism for Session-based Recommendation"

active-preference-based-gpr

Companion code for RSS 2020 paper: "Active Preference-Based Gaussian Process Regression for Reward Learning"

activeweasul

adahessian

ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning

ahgp

[NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)

align_uniform

Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere.

alpha_omok

Minimal version of DeepMind AlphaZero

ance

A novel embedding training algorithm leveraging ANN search and achieved SOTA retrieval on Trec DL 2019 and OpenQA benchmarks

astra

Self-training with Weak Supervision (NAACL 2021)

autoattention

Tensorflow implementaion for AutoAttention

autofaiss

Automatically create Faiss knn indices with the most optimal similarity search parameters.

autofis

AutoFIS: Automatic Feature Interaction Selection in Factorization Models for Click-Through Rate Prediction

automatic-rule-induction

baiduspider

百度百科爬虫

bayesianoptimization

A Python implementation of global optimization with gaussian processes.

bert

TensorFlow code and pre-trained models for BERT

bert-multi-gpu

Feel free to fine tune large BERT models with Multi-GPU and FP16 support.

bertviz

Tool for visualizing attention in the Transformer model (BERT, GPT-2, Albert, XLNet, RoBERTa, CTRL, etc.)

bleurt

BLEURT is a metric for Natural Language Generation based on transfer learning.

blob

Source code for our paper "BLOB: a probabilistic model for recommendation that combines organic and bandit signals" published at KDD 2020.

bolt

10x faster matrix and vector operations

bootstrap_dqn

Implementation of Bootstrap DQN and Randomized Prior Functions on ALE

bpt

Source code of paper "BP-Transformer: Modelling Long-Range Context via Binary Partitioning"

budget_constrained_bidding

Budget Constrained Bidding for Display Advertising using Model-free Reinforcement Learning

byteps

A high performance and generic framework for distributed DNN training

calm-dialogue

capsnet-tensorflow

A Tensorflow implementation of CapsNet(Capsules Net) in paper Dynamic Routing Between Capsules

chmm-alt

Code for "BERTifying the Hidden Markov Model for Multi-Source Weakly Supervised Named Entity Recognition"

cikm2020-s3rec

Code for CIKM2020 "S3-Rec: Self-Supervised Learning for Sequential Recommendation with Mutual Information Maximization"

clsr

The official implementation of "Disentangling Long and Short-Term Interests for Recommendation" (WWW '22)