claudecode/AvianVisitors
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixes

Browse Source
This commit is contained in:
ehpersonal38
2023-01-16 13:21:44 -05:00
parent 2685171e75
commit f03f69cc63
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/species.py
+3 -3
View File
@@ -71,13 +71,13 @@ def predictFilter(lat, lon, week):
return M_INTERPRETER.get_tensor(M_OUTPUT_LAYER_INDEX)[0]
def explore(lat, lon, week):
def explore(lat, lon, week, threshold):
# Make filter prediction
l_filter = predictFilter(lat, lon, week)
# Apply threshold
l_filter = np.where(l_filter >= 0.03, l_filter, 0)
l_filter = np.where(l_filter >= threshold, l_filter, 0)
# Zip with labels
l_filter = list(zip(l_filter, CLASSES))
@@ -92,7 +92,7 @@ def getSpeciesList(lat, lon, week, threshold=0.05, sort=False):
print('Getting species list for {}/{}, Week {}...'.format(lat, lon, week), end='', flush=True)
# Extract species from model
pred = explore(lat, lon, week)
pred = explore(lat, lon, week, threshold)
# Make species list
slist = []