Skip to content

Commit

Permalink
0.0.9 release
Browse files Browse the repository at this point in the history
  • Loading branch information
gallettilance committed Nov 15, 2024
1 parent 8a9bda4 commit 69ff2ba
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions kviz/visualizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -342,10 +342,10 @@ def fit(self, X, Y, snap_freq=10, filename='decision_boundary', duration=1000, *

for _ in range(int(epochs / snap_freq)):
self.model.fit(X, Y, epochs=snap_freq, **kwargs)
self._int_models = self._get_int_models()
# self._int_models = self._get_int_models() TODO: make this function more efficient
images.append(im.fromarray(self._snap_decision_boundary(X, Y, filename)))

self._convert_gif(images, filename + 'gif', duration)
self._convert_gif(images, filename, duration)
return self.model


Expand Down Expand Up @@ -425,7 +425,7 @@ def view_activations_for(self, X, filename='activations', duration=1000, x_color
input_images.append(self._snap_X([i], X, filename, x_color=x_color, x_marker=x_marker))
self._reset()

self._convert_gif(network_images, filename, duration)
self._stack_gifs(network_images, input_images, filename, duration)
return


Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@
setup(
name='kviz',
packages=['kviz'],
version='0.0.8',
version='0.0.9',
description='A Library for visualizing keras neural networks',
install_requires=[
'tensorflow',
Expand Down

0 comments on commit 69ff2ba

Please sign in to comment.