Skip to content

A minimal implementation of Data Maps for the Huggingface libaries from the "Dataset Cartography: Mapping and Diagnosing Datasets with Training Dynamics" paper

License

Notifications You must be signed in to change notification settings

JP-SystemsX/tiny_data_maps

Repository files navigation

Tiny Data Maps

A minimal implementation of Data Maps for the Huggingface libraries from the "Dataset Cartography: Mapping and Diagnosing Datasets with Training Dynamics" paper.

In a Nutshell

This project contains callbacks that watch training dynamics (i.e. how well each gold label could be predicted after each epoch).
This allows the creation of dataset maps as the one below that helps to find out which samples of a dataset are easier and which are harder to learn.
An example on how to use it is given in the main.py file

img_1.png

Sources

A more complete version (i.e. the original implementation) can be found here and a very good implementation as a tensorflow callback that served as template for this project can be found here under learning.DataMapCallback

About

A minimal implementation of Data Maps for the Huggingface libaries from the "Dataset Cartography: Mapping and Diagnosing Datasets with Training Dynamics" paper

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages