Skip to content

Conversation

@singjc
Copy link
Owner

@singjc singjc commented May 28, 2025

This pull request introduces several enhancements and updates to the ReDeeM project, focusing on adding CUDA support, improving data handling, and extending model capabilities. Key changes include the addition of a CUDA-enabled Dockerfile, updates to data handling with metadata support, and the introduction of new models in the redeem-properties crate. Below is a categorized summary of the most important changes:

Infrastructure and Build Updates:

  • Added a new Dockerfile to enable building the project with NVIDIA CUDA support, setting up the environment for GPU-accelerated computations.
  • Updated the Cargo.toml workspace to include the new redeem-cli crate.

Data Handling Enhancements:

  • Introduced a new PsmMetadata struct in data_handling.rs to store metadata such as spectrum IDs, file IDs, and feature names. This metadata is now integrated into the Experiment struct. [1] [2] [3]
  • Refactored example scripts (gbdt_semi_supervised_learning.rs, svm_semi_supervised_learning.rs, xgb_semi_supervised_learning.rs) to use a new load_test_psm_csv function for loading data with metadata, replacing older TSV-based methods. [1] [2] [3]

Model and Feature Updates:

  • Expanded the list of models in the redeem-properties crate, adding CNN-Transformer-based models for RT and CCS prediction.
  • Updated model names in the README.md for consistency with the new architecture.

Code Refactoring:

  • Replaced the license field in redeem-classifiers/Cargo.toml with a blank entry, possibly for future updates.
  • Enhanced logging and error handling in the example scripts to improve debugging and maintainability. [1] [2] [3]

These changes collectively improve the project's usability, scalability, and support for advanced features like GPU acceleration and metadata-driven workflows.

singjc and others added 30 commits May 7, 2025 10:07
Add Clone marco to Parameters struct
feat: Add new modules for training and loading data in redeem-cli
singjc added 29 commits May 13, 2025 21:06
@singjc singjc merged commit 9b918b7 into master May 28, 2025
1 check passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants