diff --git a/IMDB/02-Prediction.ipynb b/IMDB/02-Prediction.ipynb index f9474b9fd7929b1897389111a29468d4c53517cd..4c9c3c2b7912c654dfd8a07be511e2c054c1bebe 100644 --- a/IMDB/02-Prediction.ipynb +++ b/IMDB/02-Prediction.ipynb @@ -138,7 +138,10 @@ "metadata": {}, "outputs": [], "source": [ - "with open('./data/word_index.json', 'r') as fp:\n", + "place, dataset_dir = ooo.good_place( { 'GRICAD' : f'{os.getenv(\"SCRATCH_DIR\",\"\")}/PROJECTS/pr-fidle/datasets/IMDB',\n", + " 'IDRIS' : f'{os.getenv(\"WORK\",\"\")}/datasets/IMDB',\n", + " 'HOME' : f'{os.getenv(\"HOME\",\"\")}/datasets/IMDB'} )\n", + "with open(dataset_dir+'/word_index.json', 'r') as fp:\n", " word_index = json.load(fp)\n", " index_word = {index:word for word,index in word_index.items()} " ]