josephj6802/codereviewer

Codereviewer ONNX Model

This is an ONNX version of the microsoft/codereviewer model.

Model Description

The Codereviewer model is designed for text classification tasks, specifically for code review comments. This model has been converted to the ONNX format to enable efficient inference.

Intended Use

This model can be used to classify code review comments into various categories. It is suitable for integrating into automated code review systems to assist developers in identifying and categorizing comments.

How to Use

You can use the model with the transformers library by loading it as follows:

from transformers import pipeline

model_id = "josephj6802/codereviewer"

# Load the model
classifier = pipeline("text-classification", model=model_id)

# Use the model for inference
result = classifier("Your input text here")
print(result)

Citation

@article{li2022codereviewer,
  title={CodeReviewer: Pre-Training for Automating Code Review Activities},
  author={Li, Zhiyu and Lu, Shuai and Guo, Daya and Duan, Nan and Jannu, Shailesh and Jenks, Grant and Majumder, Deep and Green, Jared and Svyatkovskiy, Alexey and Fu, Shengyu and others},
  journal={arXiv preprint arXiv:2203.09095},
  year={2022}
}