annotation.api/src/main/java/ch/benlegiang/annotation/api/entities/AnnotationEntity.java
<gh_stars>0package ch.benlegiang.annotation.api.entities;

import ch.benlegiang.annotation.api.enums.CodeLanguage;
import lexer.LTok;
import lombok.Data;
import org.springframework.data.annotation.Id;
import org.springframework.data.annotation.Transient;
import org.springframework.data.mongodb.core.mapping.Document;
import java.util.List;

@Data
@Document(collection = "annotations")
public class AnnotationEntity {
    @Id
    public String id;
    public CodeLanguage codeLanguage;
    public String sourceCode;
    @Transient
    public LTok[] tokens;
    @Transient
    public List<Integer> lexed;
    public List<Integer> hCodeTokenIds;
    public List<Integer> hCodeValues;
    @Transient
    public List<Integer> prediction;
    public Boolean isTrained = false;
    public Boolean isValidated = false;
    public Boolean correctPrediction;
}
