Skip to content
Snippets Groups Projects
Commit b607030c authored by Nikolai.Hartmann's avatar Nikolai.Hartmann
Browse files

renaming tfhelpers to utils

parent 7058afbf
No related branches found
No related tags found
No related merge requests found
...@@ -10,7 +10,7 @@ from mpl_toolkits.axes_grid1 import ImageGrid, make_axes_locatable ...@@ -10,7 +10,7 @@ from mpl_toolkits.axes_grid1 import ImageGrid, make_axes_locatable
import numpy as np import numpy as np
from .keras_visualize_activations.read_activations import get_activations from .keras_visualize_activations.read_activations import get_activations
from .tfhelpers import get_grad_function, max_activation_wrt_input, create_random_event from .utils import get_grad_function, max_activation_wrt_input, create_random_event
import logging import logging
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
...@@ -328,10 +328,9 @@ if __name__ == "__main__": ...@@ -328,10 +328,9 @@ if __name__ == "__main__":
import logging import logging
logging.basicConfig() logging.basicConfig()
logging.getLogger("tfhelpers").setLevel(logging.DEBUG) logging.getLogger().setLevel(logging.DEBUG)
logging.getLogger(__name__).setLevel(logging.DEBUG)
from .tfhelpers import get_single_neuron_function, get_max_activation_events from .utils import get_single_neuron_function, get_max_activation_events
import meme import meme
# meme.setOptions(overrideCache="/scratch-local/nhartmann/meme_cache") # meme.setOptions(overrideCache="/scratch-local/nhartmann/meme_cache")
......
...@@ -15,7 +15,7 @@ from KerasROOTClassification.plotting import ( ...@@ -15,7 +15,7 @@ from KerasROOTClassification.plotting import (
plot_hist_2D_events, plot_hist_2D_events,
plot_cond_avg_actmax_2D plot_cond_avg_actmax_2D
) )
from KerasROOTClassification.tfhelpers import get_single_neuron_function, get_max_activation_events from KerasROOTClassification.utils import get_single_neuron_function, get_max_activation_events
parser = argparse.ArgumentParser(description='Create various 2D plots for a single neuron') parser = argparse.ArgumentParser(description='Create various 2D plots for a single neuron')
parser.add_argument("project_dir") parser.add_argument("project_dir")
......
...@@ -8,7 +8,7 @@ import keras.backend as K ...@@ -8,7 +8,7 @@ import keras.backend as K
from meme import cache from meme import cache
logger = logging.getLogger("tfhelpers") logger = logging.getLogger(__name__)
logger.addHandler(logging.NullHandler()) logger.addHandler(logging.NullHandler())
def get_single_neuron_function(model, layer, neuron, scaler=None): def get_single_neuron_function(model, layer, neuron, scaler=None):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment