38 lines (38 with data), 2.5 kB

3

ΈA]	γ@s@dZd
dl
Z
d
dlZ
d
dlZd
dlZGdd„de
jjjj	ƒZ
dS)zl
sampler.py
=======================
Balanced sampling based on one of the columns of the patch information.
ιNc@s2eZ
dZd
Zddd„
Zdd„Zdd„Zd	d
„ZdS)ΪImbalancedDatasetSamplera-
Samples elements randomly from a given list of indices for imbalanced dataset
    https://raw.githubusercontent.com/ufoym/imbalanced-dataset-sampler/master/sampler.py
    Arguments:
        indices (list, optional): a list of indices
        num_samples (int, optional): number of samples to draw
    Ncs¬|dkrtt
tˆƒ
ƒ
ƒ
n|ˆ_tˆjƒ
ˆ_|dkr<tˆjƒ
n|ˆ_i‰
x<ˆjD]2}ˆjˆ|ƒ}|ˆ
krxˆ
|d
7<qNd
ˆ
|<qNW‡‡
‡fdd„ˆjDƒ
}tj	|ƒ
ˆ_
dS)Nι
c
s g|]}
dˆ
ˆjˆ|
ƒ‘qS)
gπ?)
Ϊ
_get_label)Ϊ.0Ϊidx)ΪdatasetΪlabel_to_countΪself©ϊC/Users/joshualevy/Documents/GitHub/PathFlowAI/pathflowai/sampler.pyϊ
<listcomp>-s
z5ImbalancedDatasetSampler.__init__.<locals>.<listcomp>)ΪlistΪrangeΪlenΪindicesΪtargetsΪ	n_targetsΪnum_samplesrΪtorchΪDoubleTensorΪweights)r	rrrrΪlabelrr
)rrr	rΪ__init__s






z!ImbalancedDatasetSampler.__init__cCsŠt|
ƒ
}|t
jjkr"|
j|jƒS|t
jjkr<|
j|d
S|
jj	||
j
j}|jd
krft
j|ƒ
}nt|tt
jfƒr~|d}t|ƒ
SdS)Nrr
)ΪtypeΪtorchvisionΪdatasetsΪMNISTΪtrain_labelsΪitemΪImageFolderΪimgsΪ
patch_infoΪilocrΪvaluesrΪnpΪargmaxΪ
isinstancer
ΪndarrayΪint)r	rrZdataset_typeΪ
yr
r
rr1s









z#ImbalancedDatasetSampler._get_labelc

s"‡f
d
d„tj
ˆjˆjddDƒ
S)Nc
3s|]}
ˆj|
V
qdS)
N)
r)rΪ
i)
r	r
rϊ	<genexpr>Asz4ImbalancedDatasetSampler.__iter__.<locals>.<genexpr>T)
Ϊreplacement)rΪmultinomialrr)
r	r
)
r	rΪ__iter__@s

z!ImbalancedDatasetSampler.__iter__c


Cs|jS)
N)
r)
r	r
r
rΪ__len__Ds
z ImbalancedDatasetSampler.__len__)NN)Ϊ__name__Ϊ
__module__Ϊ__qualname__Ϊ__doc__rrr.r/r
r
r
rr
s

r)r3rΪtorch.utils.datarΪnumpyr$ΪutilsΪdataΪsamplerΪSamplerrr
r
r
rΪ<module>s