Update README.md

README.md

@@ -4,13 +4,13 @@ tags:
 - multimodal
 - classification
 datasets:
-- MeriDK/AstroM3Processed
-- MeriDK/AstroM3Dataset
+- AstroMLCore/AstroM3Processed
+- AstroMLCore/AstroM3Dataset
 ---
 
 AstroM³ is a self-supervised multimodal model for astronomy that integrates time-series photometry, spectra, and metadata into a unified embedding space
-for classification and other downstream tasks. AstroM³ is trained on [AstroM3Processed](https://huggingface.co/datasets/AstroFOMO/AstroM3Processed), 
-which is the pre-processed version of [AstroM3Dataset](https://huggingface.co/datasets/AstroFOMO/AstroM3Dataset).
+for classification and other downstream tasks. AstroM³ is trained on [AstroM3Processed](https://huggingface.co/datasets/AstroMLCore/AstroM3Processed), 
+which is the pre-processed version of [AstroM3Dataset](https://huggingface.co/datasets/AstroMLCore/AstroM3Dataset).
 For more details on the AstroM³ architecture, training, and results, please refer to the [paper](https://arxiv.org/abs/2411.08842).
 
 <p align="center">
@@ -43,7 +43,7 @@ from datasets import load_dataset
 from src.data import process_photometry
 
 # Load the test dataset
-test_dataset = load_dataset('AstroFOMO/AstroM3Processed', name='full_42', split='test')
+test_dataset = load_dataset('AstroMLCore/AstroM3Processed', name='full_42', split='test')
 
 # Process photometry to have a fixed sequence length of 200 (center-cropped)
 test_dataset = test_dataset.map(process_photometry, batched=True, fn_kwargs={'seq_len': 200, 'how': 'center'})
@@ -55,7 +55,7 @@ import torch
 from src.model import AstroM3
 
 # Load the base AstroM3-CLIP model
-model = AstroM3.from_pretrained('AstroFOMO/AstroM3-CLIP')
+model = AstroM3.from_pretrained('AstroMLCore/AstroM3-CLIP')
 
 # Retrieve the first sample (batch size = 1)
 sample = test_dataset[0:1]
@@ -80,22 +80,22 @@ print('Multimodal Embedding (Spectra Missing):', multimodal_emb_missing)
 from src.model import AstroM3, Informer, GalSpecNet, MetaModel
 
 # Photometry classification
-photo_model = Informer.from_pretrained('AstroFOMO/AstroM3-CLIP-photo')
+photo_model = Informer.from_pretrained('AstroMLCore/AstroM3-CLIP-photo')
 prediction = photo_model(photometry, photometry_mask).argmax(dim=1).item()
 print('Photometry Classification:', test_dataset.features['label'].int2str(prediction))
 
 # Spectra classification
-spectra_model = GalSpecNet.from_pretrained('AstroFOMO/AstroM3-CLIP-spectra')
+spectra_model = GalSpecNet.from_pretrained('AstroMLCore/AstroM3-CLIP-spectra')
 prediction = spectra_model(spectra).argmax(dim=1).item()
 print('Spectra Classification:', test_dataset.features['label'].int2str(prediction))
 
 # Metadata classification
-meta_model = MetaModel.from_pretrained('AstroFOMO/AstroM3-CLIP-meta')
+meta_model = MetaModel.from_pretrained('AstroMLCore/AstroM3-CLIP-meta')
 prediction = meta_model(metadata).argmax(dim=1).item()
 print('Metadata Classification:', test_dataset.features['label'].int2str(prediction))
 
 # Multimodal classification
-all_model = AstroM3.from_pretrained('AstroFOMO/AstroM3-CLIP-all')
+all_model = AstroM3.from_pretrained('AstroMLCore/AstroM3-CLIP-all')
 prediction = all_model(photometry, photometry_mask, spectra, metadata).argmax(dim=1).item()
 print('Multimodal Classification:', test_dataset.features['label'].int2str(prediction))
 ```
@@ -104,11 +104,11 @@ print('Multimodal Classification:', test_dataset.features['label'].int2str(predi
 
 | # Model | # Description |
 | :--- | :--- |
-| [AstroM3-CLIP](https://huggingface.co/AstroFOMO/AstroM3-CLIP) | The base model pre-trained using the trimodal CLIP approach. |
-| [AstroM3-CLIP-meta](https://huggingface.co/AstroFOMO/AstroM3-CLIP-meta) | Fine-tuned for metadata-only classification. |
-| [AstroM3-CLIP-spectra](https://huggingface.co/AstroFOMO/AstroM3-CLIP-spectra) | Fine-tuned for spectra-only classification. |
-| [AstroM3-CLIP-photo](https://huggingface.co/AstroFOMO/AstroM3-CLIP-photo) | Fine-tuned for photometry-only classification. |
-| [AstroM3-CLIP-all](https://huggingface.co/AstroFOMO/AstroM3-CLIP-all) | Fine-tuned for multimodal classification. |
+| [AstroM3-CLIP](https://huggingface.co/AstroMLCore/AstroM3-CLIP) | The base model pre-trained using the trimodal CLIP approach. |
+| [AstroM3-CLIP-meta](https://huggingface.co/AstroMLCore/AstroM3-CLIP-meta) | Fine-tuned for metadata-only classification. |
+| [AstroM3-CLIP-spectra](https://huggingface.co/AstroMLCore/AstroM3-CLIP-spectra) | Fine-tuned for spectra-only classification. |
+| [AstroM3-CLIP-photo](https://huggingface.co/AstroMLCore/AstroM3-CLIP-photo) | Fine-tuned for photometry-only classification. |
+| [AstroM3-CLIP-all](https://huggingface.co/AstroMLCore/AstroM3-CLIP-all) | Fine-tuned for multimodal classification. |
 
 ## AstroM3-CLIP Variants
 These variants of the base AstroM3-CLIP model are trained using different random seeds (42, 0, 66, 12, 123); 
@@ -116,11 +116,11 @@ ensure that the dataset is loaded with the corresponding seed for consistency.
 
 | # Model | # Description |
 | :--- | :--- |
-| [AstroM3-CLIP-42](https://huggingface.co/AstroFOMO/AstroM3-CLIP-42) | The base model pre-trained with random seed 42 (identical to AstroM3-CLIP). |
-| [AstroM3-CLIP-0](https://huggingface.co/AstroFOMO/AstroM3-CLIP-0) | AstroM3-CLIP pre-trained with random seed 0 (use dataset with seed 0). |
-| [AstroM3-CLIP-66](https://huggingface.co/AstroFOMO/AstroM3-CLIP-66) | AstroM3-CLIP pre-trained with random seed 66 (use dataset with seed 66). |
-| [AstroM3-CLIP-12](https://huggingface.co/AstroFOMO/AstroM3-CLIP-12) | AstroM3-CLIP pre-trained with random seed 12 (use dataset with seed 12). |
-| [AstroM3-CLIP-123](https://huggingface.co/AstroFOMO/AstroM3-CLIP-123) | AstroM3-CLIP pre-trained with random seed 123 (use dataset with seed 123). |
+| [AstroM3-CLIP-42](https://huggingface.co/AstroMLCore/AstroM3-CLIP-42) | The base model pre-trained with random seed 42 (identical to AstroM3-CLIP). |
+| [AstroM3-CLIP-0](https://huggingface.co/AstroMLCore/AstroM3-CLIP-0) | AstroM3-CLIP pre-trained with random seed 0 (use dataset with seed 0). |
+| [AstroM3-CLIP-66](https://huggingface.co/AstroMLCore/AstroM3-CLIP-66) | AstroM3-CLIP pre-trained with random seed 66 (use dataset with seed 66). |
+| [AstroM3-CLIP-12](https://huggingface.co/AstroMLCore/AstroM3-CLIP-12) | AstroM3-CLIP pre-trained with random seed 12 (use dataset with seed 12). |
+| [AstroM3-CLIP-123](https://huggingface.co/AstroMLCore/AstroM3-CLIP-123) | AstroM3-CLIP pre-trained with random seed 123 (use dataset with seed 123). |
 
 ## Using your own data
 
@@ -132,5 +132,5 @@ If you want to use the model with your own data, you must pre-process it in the
 3. **Metadata**: Scalar metadata is transformed via domain-specific functions (e.g., absolute magnitude, log, sin/cos), then normalized using dataset-level statistics.
 
 For a detailed description, read the [paper](https://arxiv.org/abs/2411.08842).  
-To see exactly how we performed this preprocessing, refer to [`preprocess.py`](https://huggingface.co/datasets/AstroFOMO/AstroM3Dataset/blob/main/preprocess.py) in the AstroM3Dataset repo.
+To see exactly how we performed this preprocessing, refer to [`preprocess.py`](https://huggingface.co/datasets/AstroMLCore/AstroM3Dataset/blob/main/preprocess.py) in the AstroM3Dataset repo.