-
Notifications
You must be signed in to change notification settings - Fork 0
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
csv output (headers), partial fix for PB freezing notebook, skipping reshaping #10
base: main
Are you sure you want to change the base?
Conversation
@@ -250,8 +250,9 @@ def normalizeAgents(agents): | |||
dataForNormalization = [] | |||
pb = ProgressBar(len(agents)) | |||
pb.start() | |||
for agent in agents: | |||
pb.check(0, True) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I suspect this fixes some freezing since instead of printing for each loop we are using the underlying chunked updated of the PB class
|
||
|
||
# In[]: | ||
|
||
|
||
def reshapeFlattenedTrainingSets(allTrainingSetsFlattened_X, allTrainingSetsFlattened_Y): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Removed reshaping function since the previous step now outputs a numpy array so we don't have to reshape the list to a numpy array
|
||
|
||
# In[]: | ||
|
||
|
||
allTrainingSetsFlattened_X, allTrainingSetsFlattened_Y = flattenTrainingSets(allTrainingSets) | ||
def flattenTrainingSets(allTrainingSets, totalNumberOfTrainingSets): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
rewrote this function so that it outputs a numpy array so we don't have to reshape the list later on
464d30c
to
00d31a8
Compare
since test sets will only need to be the last `n` history frames of an agent
currently working on getting testing part 1. getting copy of mask already 2. added ability to binary search backwards a mask index
#5