CS584 源码

作业1-说明 第1步:创建嵌入 (选项1)运行文件doc2vec-encoder.py (选项2)运行文件句子transformers.py 注意:这两个文件都会在根目录中生成两个名为x_train.pkl和x_test.pkl的文件。 步骤2:做出预测 运行文件knn.py 注意:此文件将在根目录中生成预测文件。 步骤3(可选):精调句子转换器模型 运行文件fine_tune.py 运行文件knn.py-但请确保已从根目录中的微调模型文件夹加载了模型 注意:此文件将产生一个名为x_train.pkl的文件,并将模型保存在根目录中。