Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Create separate files for each deduplication class #389

Closed
wants to merge 5 commits into from
Closed

Create separate files for each deduplication class #389

wants to merge 5 commits into from

Conversation

sarahyurick
Copy link
Collaborator

No description provided.

@sarahyurick
create separate fuzzy dedup files
6d0771a 
Signed-off-by: Sarah Yurick <[email protected]>
@sarahyurick
fix import paths
275a20a 
Signed-off-by: Sarah Yurick <[email protected]>
@sarahyurick
add semantic dedup
8975e78 
Signed-off-by: Sarah Yurick <[email protected]>
@sarahyurick
run black and isort
9ead4b9 
Signed-off-by: Sarah Yurick <[email protected]>
@sarahyurick sarahyurick added the gpuci Run GPU CI/CD on PR label Nov 22, 2024
@sarahyurick sarahyurick changed the title Split deduplication classes into multiple files Create separate files for each deduplication class Nov 22, 2024
@sarahyurick
merge conflict
28c3ff8 
Signed-off-by: Sarah Yurick <[email protected]>
sarahyurick added a commit to sarahyurick/NeMo-Curator that referenced this pull request Dec 3, 2024
@sarahyurick
add changes from NVIDIA#389
6846945 
Signed-off-by: Sarah Yurick <[email protected]>
@sarahyurick sarahyurick mentioned this pull request Dec 3, 2024
Open
@sarahyurick
Copy link
Collaborator Author

Closing in favor of #409. Files affected (26 total):

  • nemo_curator/modules/init.py
  • nemo_curator/modules/fuzzy_dedup.py
  • nemo_curator/modules/fuzzy_dedup/_mapbuckets.py
  • nemo_curator/modules/fuzzy_dedup/_shuffle.py
  • nemo_curator/modules/fuzzy_dedup/bucketstoedges.py
  • nemo_curator/modules/fuzzy_dedup/connectedcomponents.py
  • nemo_curator/modules/fuzzy_dedup/fuzzyduplicates.py
  • nemo_curator/modules/fuzzy_dedup/jaccard_similarity.py
  • nemo_curator/modules/fuzzy_dedup/lsh.py
  • nemo_curator/modules/fuzzy_dedup/minhash.py
  • nemo_curator/modules/semantic_dedup/clusteringmodel.py
  • nemo_curator/modules/semantic_dedup/embeddings.py
  • ? nemo_curator/modules/semantic_dedup/semantic_dedup.py
  • nemo_curator/modules/semantic_dedup/semanticclusterleveldedup.py
  • nemo_curator/modules/semantic_dedup/semdedup.py
  • nemo_curator/scripts/semdedup/clustering.py
  • nemo_curator/scripts/semdedup/compute_embeddings.py
  • nemo_curator/scripts/semdedup/extract_dedup_data.py
  • nemo_curator/scripts/fuzzy_deduplication/connected_components.py
  • nemo_curator/scripts/fuzzy_deduplication/jaccard_compute.py
  • nemo_curator/scripts/fuzzy_deduplication/jaccard_shuffle.py
  • nemo_curator/scripts/fuzzy_deduplication/map_buckets.py
  • tutorials/pretraining-data-curation/red-pajama-v2-curation-tutorial.ipynb
  • tutorials/single_node_tutorial/single_gpu_tutorial.ipynb
  • tutorials/zyda2-tutorial/1_fuzzy_dedup/2_buckets_to_edges.py
  • tutorials/zyda2-tutorial/1_fuzzy_dedup/3_connected_components.py

@sarahyurick sarahyurick closed this Dec 3, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@VibhuJawa VibhuJawa Awaiting requested review from VibhuJawa

@ayushdg ayushdg Awaiting requested review from ayushdg

Assignees
No one assigned
Labels
gpuci Run GPU CI/CD on PR
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@sarahyurick