问题描述
这是在数据增强过程中编写的代码。
from imblearn.over_sampling import RandomOverSampler
from imblearn.keras import balanced_batch_generator
from tensorflow.keras.utils import Sequence
class BalancedDataGenerator(Sequence):
"""ImageDataGenerator + RandomOversampling"""
def __init__(self,x,y,datagen,batch_size=256):
self.datagen = datagen
self.batch_size = batch_size
self._shape = x.shape
datagen.fit(x)
self.gen,self.steps_per_epoch = balanced_batch_generator(x.reshape(x.shape[0],-1),sampler=RandomOverSampler(),batch_size=self.batch_size,keep_sparse=True)
def __len__(self):
return self._shape[0] // self.batch_size
def __getitem__(self,idx):
x_batch,y_batch = self.gen.__next__()
x_batch = x_batch.reshape(-1,*self._shape[1:])
return self.datagen.flow(x_batch,y_batch,batch_size=self.batch_size).next()
但是,在使用 imblearn 时,我遇到了以下错误消息:- 类型错误:重复基类序列
请帮忙
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)