Franblueee/torchmil

This library helps machine learning engineers and researchers build and train deep learning models for problems where data is grouped into 'bags' of instances rather than individual data points. You input datasets organized this way, like medical images composed of many small patches, and it outputs trained models capable of classifying or analyzing these 'bags'. This is useful for anyone working with specialized machine learning tasks that require Multiple Instance Learning.

Available on PyPI.