natelgrw commited on
Commit
68313f5
·
1 Parent(s): 8401b89

readme commit

Browse files
Files changed (1) hide show
  1. README.md +7 -6
README.md CHANGED
@@ -10,15 +10,16 @@ tags:
10
 
11
  A collection of machine learning models for predicting maximum absorption wavelength (λ_max) of chemical compounds in various solvents. These models use molecular descriptors to predict spectroscopic properties, useful for drug discovery, materials science, and computational chemistry applications.
12
 
 
 
13
  ## 🤖 Available Models
14
 
15
- | Model | Framework | Architecture | Score | MAE (nm) | RMSE (nm) | Status |
16
- |-------|-----------|--------------|----------|----------|-----------|---------|
17
- | **AMAX_XGB1** | XGBoost | Gradient Boosting (500 estimators) | 0.9084 | 17.682 | 35.507 | Active |
18
- | **AMAX_RF1** | Scikit-Learn | Random Forest (500 trees) | 0.9035 | 18.601 | 36.441 | Active |
19
- | **AMAX_MLP1** | PyTorch | Sequential NN (1024 → 512) | 0.8913 | 23.956 | 38.680 | Active |
20
 
21
- All models utilize **312 RDKit molecular descriptors** combining both compound and solvent features, trained on a random data split of **32,010 training samples** with **4,001 validation** and **4,002 test samples**. Each model has been retrained to eliminate data leakage and ensure robust performance evaluation.
22
 
23
  ## 📄 Citation
24
 
 
10
 
11
  A collection of machine learning models for predicting maximum absorption wavelength (λ_max) of chemical compounds in various solvents. These models use molecular descriptors to predict spectroscopic properties, useful for drug discovery, materials science, and computational chemistry applications.
12
 
13
+ Version: **1.0.0**
14
+
15
  ## 🤖 Available Models
16
 
17
+ | Model | Architecture | Overall RMSE (nm) | Overall MAE (nm) | Overall R<sup>2</sup> |
18
+ |-----|-----|-----|-----|-----|
19
+ | AMAX_XGB1 | XGBoost | 182.81 | 119.30 | 0.659 |
20
+ | AMAX_MLP1 | PyTorch Sequential MLP | 64.152 | 44.388 | 0.669 |
 
21
 
22
+ All models were evaluated across rigorous scaffold, cluster, and method splits.
23
 
24
  ## 📄 Citation
25