-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
76 lines (61 loc) · 2.09 KB
/
config.py
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
# Pranav Minasandra
# pminasandra.github.io
# November 27, 2023
"""
Configuration for rest of code.
"""
import os
import os.path
# Directories
with open(".cw", "r") as fd:
PROJECTROOT = fd.read().rstrip()
DATA = os.path.join(PROJECTROOT, "Data")
ACC_GPS_DIR = os.path.join(DATA, "ACC_GPS")
AUDIT_DIR = os.path.join(DATA, "Audits")
PREDICTIONS_DIR = os.path.join(DATA, "Predictions")
FIGURES = os.path.join(PROJECTROOT, "Figures")
# Deployments
DEPLOYMENTS = ["NQ_2021_1", "RW_2021_1", "ZU_2021_1", "ZU_2021_2"]
# Loading names of all individuals
INDIVIDUALS = {}
for dplment in DEPLOYMENTS:
dirname = os.path.join(ACC_GPS_DIR, dplment)
if dplment not in INDIVIDUALS:
INDIVIDUALS[dplment] = []
for filename in os.listdir(dirname):
INDIVIDUALS[dplment].append(filename.split('_')[1])
for dplment in INDIVIDUALS:
INDIVIDUALS[dplment] = list(set(INDIVIDUALS[dplment]))
# Behavior related config
BEHAVIORS = ["No observation", "Foraging", "Scrabbling", "Standing quadrupedal vigilance",
"Bipedal vigilance", "Sitting vigilance", "Self groom", "Social",
"Haunch quadrupedal vigilance", "Walking", "Running", "Processing",
"Reforage", "Others", "Marking"]
DROP_MISSING = True
COMBINE_BEHAVIORS = True
DROP_OTHERS = True
BEHAVIOR_SIMPLIFIER = {
'Vigilance': 'Vigilance',
'Sitting vigilance': 'Vigilance',
'Bipedal vigilance': 'Vigilance',
'Haunch quadrupedal vigilance': 'Vigilance',
'Standing quadrupedal vigilance': 'Vigilance',
'Foraging': 'Foraging',
'Scrabbling': 'Foraging',
'Walking': 'Foraging',
'Reforage': 'Foraging',
'Others': 'Others',
'Social': 'Others',
'Processing': 'Others',
'Self groom': 'Others',
'Marking': 'Others',
'Running': 'Running'
} # comes from repeated tweaking during Amlan's project
# Other tweaks
formats=['png', 'pdf', 'svg']
# Miscellaneous
SUPPRESS_INFORMATIVE_PRINT = False
AMLAN_OFFSET = -4 # There is a strange inexplicable offset between the ACC and
# the official labels
LOG_TRANSFORM_VEDBA = True
SCALE_DATA = False # turned out to be a bad idea