config.py

import torch
from transformers import set_seed
from pathlib import Path
import os


    

SMELLS = [
    "BuiltInBehavior",
    "ExtraUsageOfArrays",
    "ImproperDataStructure",
    "LackingBreak",
    "NotCaching",
    "OveruseOfLoops",
    "UnnecessaryConditionals",
    "UnnecessaryRecursion",
    "UnnecessaryTypeConversion",
]


# Configurations
SEED_NUM = 42
DROPOUT_PROB = 0.1
HIDDEN_SIZE = 768
NUM_LABELS = 9
LAST_NUM_NEURON = 9
BEST_THRESHOLD = 0.3
PRETRAINED_MODEL = "microsoft/graphcodebert-base"
MODEL_SAVE_DIR = Path(__file__).parents[1] / "models_weights"
WEIGHTS_FILE_NAME = "weights.bin"
MODEL_SAVE_PATH = MODEL_SAVE_DIR / WEIGHTS_FILE_NAME
HF_REPO_NAME = f"mjavadmt/EnergySmell"

torch.manual_seed(SEED_NUM)
set_seed(SEED_NUM)