dsrestrepo commited on
Commit
cbd92e0
1 Parent(s): 5e6c7b5

Push Keras model using huggingface_hub. (#3)

Browse files

- Push Keras model using huggingface_hub. (5acff0de59dc902eabe3650a3f80bb55e4f769d4)

README.md CHANGED
@@ -4,22 +4,17 @@ library_name: keras
4
 
5
  ## Model description
6
 
7
- Autoencoder model trained to compress information from sentinel-2 satellite images using a Vision Transformer (ViT) as encoder backbone to extract features.
8
- The latent space of the model is given by 1024 neurons which can be used to generate embeddings from the sentinel-2 satellite images.
9
-
10
- The model was trained using bands RGB (2, 3 and 4) (Red, Green and Blue) of the Sentinel-2 satellites and using 81 municipalities of Colombia with most dengue cases.
11
-
12
- The input shape of the model is 224, 224, 3. To extract features you should remove the last layer.
13
 
14
  ## Intended uses & limitations
15
 
16
- The model was trained with images of 81 different cities in Colombia, however it may require fine tuning or retraining to learn from other contexts such as countries and other continents.
17
 
18
  ## Training and evaluation data
19
 
20
- The model was trained with satellite images of 81 different cities in Colombia extracted from sentinel-2 using RGB bands using an asymmetric autoencoder. Images with information that could result in noise such as black images were filtered prior to training to avoid noise in the data.
21
 
22
- The dataset was split into train and test using 80% for train and 20% to test.
23
 
24
  ### Training hyperparameters
25
 
@@ -28,10 +23,10 @@ The following hyperparameters were used during training:
28
  | Hyperparameters | Value |
29
  | :-- | :-- |
30
  | name | Adam |
31
- | learning_rate | 0.0010000000474974513 |
32
  | decay | 0.0 |
33
- | beta_1 | 0.8999999761581421 |
34
- | beta_2 | 0.9990000128746033 |
35
  | epsilon | 1e-07 |
36
  | amsgrad | False |
37
  | training_precision | float32 |
 
4
 
5
  ## Model description
6
 
7
+ More information needed
 
 
 
 
 
8
 
9
  ## Intended uses & limitations
10
 
11
+ More information needed
12
 
13
  ## Training and evaluation data
14
 
15
+ More information needed
16
 
17
+ ## Training procedure
18
 
19
  ### Training hyperparameters
20
 
 
23
  | Hyperparameters | Value |
24
  | :-- | :-- |
25
  | name | Adam |
26
+ | learning_rate | 0.001 |
27
  | decay | 0.0 |
28
+ | beta_1 | 0.9 |
29
+ | beta_2 | 0.999 |
30
  | epsilon | 1e-07 |
31
  | amsgrad | False |
32
  | training_precision | float32 |
keras_metadata.pb CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ed6fb1d5d355843d8864c84837b68f0ce6fce818cefe8394615ce5858bf634b
3
- size 213379
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:72e9c2389aa34c4d33fc6ab68618da767284d3cbb3883ab46683ced706ad5137
3
+ size 213812
saved_model.pb CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:d534302dc73b58f3457a242c44b9210d392c590564f05d416eaafb96abf6788c
3
- size 15930729
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5701db00994f47e59cec5be0168a2fe9e2c5afb4c07a780c607d1cc34dd2990e
3
+ size 16155668
variables/variables.data-00000-of-00001 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6b45fbe8bf531f75c3e1e86e9187dd9602fd18a304586d5827d2bf7d51c96922
3
- size 761250486
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6235c13d02b9eb0a0ddf740ee5531193fce63907f54cdd88bbe0255518c0a1c2
3
+ size 761250970
variables/variables.index CHANGED
Binary files a/variables/variables.index and b/variables/variables.index differ