from TensorflowToolbox.utility import read_proto as rp from TensorflowToolbox.utility import file_io from net_flow import NetFlow import sys import yaml import json if __name__ == "__main__": if len(sys.argv) == 2: model_proto_file = sys.argv[1] else: model_proto_file = "model_proto.yaml" with open(model_proto_file, 'r') as f: model_params = yaml.load(f) print(json.dumps(model_params, indent=4)) file_io.check_exist(model_params["train_file_name"]) file_io.check_exist(model_params["test_file_name"]) net = NetFlow(model_params, True, True) net.mainloop()