Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_tf_model():
tf.reset_default_graph()
input_nodes = "inputs"
target_nodes = "preds"
meta_graph = "model_files/model.tf.meta"
# meta_graph = 'model_files/model.tf-modified.meta'
checkpoint = "model_files/model.tf"
index = "model_files/model.tf.index"
pkl_file = "model_files/const_feed_dict.pkl"
from kipoi.model import TensorFlowModel
m = TensorFlowModel(input_nodes="inputs",
target_nodes="preds",
meta_graph=meta_graph,
checkpoint=checkpoint,
const_feed_dict_pkl=pkl_file)
ops = tf.get_default_graph().get_operations()
# TODO - modify the
out = tf.train.export_meta_graph(filename='model_files/model.tf-modified.meta', as_text=True)
ops[0].outputs[0].shape[0] = None
pops = [op.outputs[0] for op in ops
if op.type == "Placeholder" and
op.name.startswith("Placeholder")]