Skip to content

Commit 66ab19d

Browse files
fix
1 parent 5562fbe commit 66ab19d

File tree

1 file changed

+1
-12
lines changed

1 file changed

+1
-12
lines changed

scripts/preprocess-remote.py

Lines changed: 1 addition & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def processFolder(folder, timeDelta, testRatio, framesPerChunk, testPadding, ski
179179
start, end, np.min(delta), np.max(delta), np.mean(delta), len(session_time), duration
180180
))
181181
stats['deltas'].append(delta)
182-
stats['durations'].append(duration)
182+
stats['durations'].append([duration])
183183
continue
184184
######################################################
185185
# split each session into training and testing sets
@@ -215,17 +215,6 @@ def saveSubset(filename, idx):
215215
return len(testing), len(training), False, stats
216216

217217
def main(args):
218-
# blacklisted datasets
219-
blacklisted = []
220-
if args.blacklist is not None:
221-
with open(args.blacklist, 'r') as f:
222-
blacklisted = json.load(f)
223-
pass
224-
blacklisted = set([
225-
'/'.join(item)
226-
for item in blacklisted
227-
])
228-
print(blacklisted)
229218
stats = {
230219
'placeId': [],
231220
'userId': [],

0 commit comments

Comments
 (0)