-
Notifications
You must be signed in to change notification settings - Fork 480
Description
Code:
import numpy as np
from sklearn.model_selection import KFold
#import tensorflow as tf
from sklearn import metrics
from sklearn.metrics import accuracy_score
from sklearn.metrics import f1_score
from sklearn.metrics import precision_score
from sklearn.metrics import recall_score
from torch.utils.data import Dataset, DataLoader,SubsetRandomSampler
from sklearn.metrics import precision_recall_fscore_support
from transformers import AdamW, get_linear_schedule_with_warmup
from openprompt import PromptDataLoader
from torch import nn
import os
error:
ModuleNotFoundError Traceback (most recent call last)
Cell In[10], line 14
12 from sklearn.metrics import precision_recall_fscore_support
13 from transformers import AdamW, get_linear_schedule_with_warmup
---> 14 from openprompt import PromptDataLoader
15 from torch import nn
16 import os
File ~/miniconda3/lib/python3.8/site-packages/openprompt/init.py:2
1 version = "1.0.1"
----> 2 from .pipeline_base import PromptDataLoader, PromptModel, PromptForClassification, PromptForGeneration
3 from .utils import *
4 from .prompt_base import Template, Verbalizer
File ~/miniconda3/lib/python3.8/site-packages/openprompt/pipeline_base.py:4
2 from torch.utils.data.sampler import RandomSampler
3 from transformers.configuration_utils import PretrainedConfig
----> 4 from transformers.generation_utils import GenerationMixin
5 import torch
6 import torch.nn as nn
ModuleNotFoundError: No module named 'transformers.generation_utils'