diff --git a/odyssey/data/bigbird_data/DataProcessor.ipynb b/odyssey/data/DataProcessor.ipynb similarity index 100% rename from odyssey/data/bigbird_data/DataProcessor.ipynb rename to odyssey/data/DataProcessor.ipynb diff --git a/odyssey/evals/AttentionVisualization.ipynb b/odyssey/interp/AttentionVisualization.ipynb similarity index 99% rename from odyssey/evals/AttentionVisualization.ipynb rename to odyssey/interp/AttentionVisualization.ipynb index dc0a91c..2a40a8f 100644 --- a/odyssey/evals/AttentionVisualization.ipynb +++ b/odyssey/interp/AttentionVisualization.ipynb @@ -64,7 +64,7 @@ "from odyssey.data.dataset import FinetuneDataset\n", "from odyssey.data.tokenizer import ConceptTokenizer\n", "from odyssey.models.prediction import load_finetuned_model, predict_patient_outcomes\n", - "from odyssey.models.utils import (\n", + "from odyssey.models.model_utils import (\n", " load_finetune_data,\n", ")" ]