文件名称:mixmatch-pytorch:用PyTorch实现MixMatch
文件大小:6KB
文件格式:ZIP
更新时间:2024-06-12 22:35:32
Python
mixmatch-pytorch 使用PyTorch实现MixMatch( ) 安装 pip install git+https://github.com/FelixAbrahamsson/mixmatch-pytorch 指示 该软件包提供了一个类mixmatch_pytorch.MixMatchLoader ,该类的工作方式与普通的PyTorch DataLoader一样,还提供了一个由mixmatch_pytorch.get_mixmatch_loss构造的损失函数。 使用示例,请参见下文。 您必须提供一个数据加载器,该数据加载器具有可重复使用的屈服字典,其键'features'和'targets'具有标记数据集的增强(!)功能和目标。 还必须为未标记的数据提供数据集,该数据集可以包装在PyTorch DataLoader中。 数据集必须返回带有'features'增强功能的
【文件预览】:
mixmatch-pytorch-master
----setup.py(359B)
----.gitignore(21B)
----requirements.txt(11B)
----README.md(2KB)
----mixmatch_pytorch()
--------get_mixmatch_loss.py(618B)
--------mixup_samples.py(211B)
--------__init__.py(93B)
--------sharpen.py(395B)
--------tile_adjacent.py(540B)
--------k_batch_sampler.py(441B)
--------guess_targets.py(584B)
--------mixmatch_batch.py(1KB)
--------mixmatch_loader.py(1KB)
--------get_unlabeled_loader.py(291B)