With this model, the return value can be classified based on the method name only. Six classes are available for this purpose. The model answers with the label indexes and the associated probability. The corresponding class name must then be taken from the following list: ['None', 'Number', 'Boolean', 'String', 'Object', 'Collection'].

The corresponding huggingface space can be found at https://huggingface.co/spaces/UDE-SE/ReturnTypePredictor.

Good luck trying it out!

Downloads last month
17
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train UDE-SE/CodeBERTForReturnTypeClassification

Space using UDE-SE/CodeBERTForReturnTypeClassification 1