
class Config:
    def __init__(self):
        # 1- 原始文件路径
        # 训练集
        self.train_datapath = "../00-data/data/train.txt"
        # 测试集
        self.test_datapath = "../00-data/data/test.txt"
        # 验证集
        self.dev_datapath = "../00-data/data/dev.txt"
        # 类别种类
        self.class_datapath = "../00-data/data/class.txt"
        # 训练集
        self.stopwords_datapath = "../00-data/data/stopwords.txt"

        # 2- 分词预处理后文件存放路径
        # 训练集
        self.process_train_datapath = "data/process_train_data.csv"
        # 测试集
        self.process_test_datapath = "data/process_test_data.csv"
        # 验证集
        self.process_dev_datapath = "data/process_dev_data.csv"

        # ------------------------ 理解：简化代码 ------------------------
        # 1- 原始文件路径
        self.original_datapath_dict = {
            "train": self.train_datapath,
            "test": self.test_datapath,
            "dev": self.dev_datapath
        }
        # 2- 分词预处理后文件存放路径
        self.process_datapath_dict = {
            "train": self.process_train_datapath,
            "test": self.process_test_datapath,
            "dev": self.process_dev_datapath
        }

        # 3- 训练好的模型保存路径
        self.tfidf_model_path = "save_model/tfidf.pkl"
        self.rf_model_path = "save_model/rf.pkl"

        # 4- 预测结果存储路径
        self.predict_result_path = "result/dev.csv"
