File size: 1,101 Bytes
2ee1fb2
c79073a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9fe11bc
c79073a
2ee1fb2
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import os
import json


with open("default_inputs.json", "r") as default_inputs_file:
    DEFAULT_INPUTS = json.load(default_inputs_file)

with open("genus_labels.json", "r") as genus_labels_file:
    GENUS_TO_ID_MAP = json.load(genus_labels_file)
    ID_TO_GENUS_MAP = {v: k for k, v in GENUS_TO_ID_MAP.items()}

ECOLAYERS = [
    "median_elevation_1km",
    "human_footprint",
    "population_density_1km",
    "annual_precipitation",
    "precipitation_seasonality",
    "annual_mean_air_temp",
    "temp_seasonality",
]

MODELS = {
    "embeddings": "LofiAmazon/BarcodeBERT-Entire-BOLD",
    "classification": "LofiAmazon/BarcodeBERT-Finetuned-Amazon",
}

DATASETS = {
    "amazon": "LofiAmazon/BOLD-Embeddings-Ecolayers-Amazon",
}

HUGGINGFACE_DW_URL = "https://huggingface.co/datasets/LofiAmazon/Global-Ecolayers/resolve/main/{filename}?download=true"

LAYER_NAMES = [
    "median_elevation_1km.tiff",
    "human_footprint.tiff",
    "population_density_1km.tif",
    "annual_precipitation.tif",
    "precipitation_seasonality.tif",
    "annual_mean_air_temp.tif",
    "temp_seasonality.tif",
]