Maschinelles Lernen · GitHub

Transformer Network for Remaining Useful Life Prediction of Lithium-Ion Batteries

基于深度学习的肿瘤辅助诊断系统，以图像分割为核心，利用人工智能完成肿瘤区域的识别勾画并提供肿瘤区域的特征来辅助医生进行诊断。有完整的模型构建、后端架设、工业级部署和前端访问功能。TensorRT、PyTorch 、OpenCV 、Flask、Vue

基于强化学习的黑神话悟空AI

ThunderGBM: Fast GBDTs and Random Forests on GPUs

A pytorch implementation of MADDPG (multi-agent deep deterministic policy gradient)

Deep learning-based outlier/anomaly detection

code and docs for the EMNLP paper "DeepPath: A Reinforcement Learning Method for Knowledge Graph Reasoning"

Visualize PyTorch tensors with a single line of code.

中文语音识别

A-share quantitative system. A股量化系统

Pytorch model to caffe model, supported pytorch 0.3, 0.3.1, 0.4, 0.4.1 ,1.0 , 1.0.1 , 1.2 ,1.3 .notice that only pytorch 1.1 have some bugs

• Converting a pytorch model to caffe model.
• Some convenient tools of manipulate caffemodel and prototxt quickly(like get or set weights of layers).

(ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"

PyTorch Implementation of MVSNet

[ICLR 2024] Official Implementation of "Diffusion-TS: Interpretable Diffusion for General Time Series Generation"

A stock trading bot that uses machine learning to make price predictions.

Label-Efficient Semantic Segmentation with Diffusion Models (ICLR'2022)

[ICML 2023] The official implementation of the paper "TabDDPM: Modelling Tabular Data with Diffusion Models"

Faster Recurrent Neural Network Language Modeling Toolkit with Noise Contrastive Estimation and Hierarchical Softmax