swap to using 600 as dataset

This commit is contained in:
Fraser
2023-03-30 10:09:43 -04:00
parent 90f274eb38
commit 3c7dbef54d
2 changed files with 1 additions and 1 deletions
+1 -1
View File
@@ -14,7 +14,7 @@ COMPLETIONS_MODEL = "gpt-3.5-turbo"
import pathlib
project_path = pathlib.Path(__file__).parent
PATH_TO_DATASET_DICT = project_path / "dataset_dict.pkl"
PATH_TO_DATASET_DICT = project_path / "dataset_dict_500.pkl"
with open(PATH_TO_DATASET_DICT, 'rb') as f:
data = pickle.load(f)