def selectModel(): A = zk.get('/selectModel/param/A') b = zk.get('/selectModel/param/b') p = zk.get('/selectModel/param/p') log.logSend("INFO device " + localhost + " get model selector parameters") p = pickle.loads(p[0]) best_predicted_arm = np.argmax(p) if best_predicted_arm == 0: ftp = FTPClient.ftpconnect("127.0.0.1", "ftp***", "Guest***") FTPClient.downloadfile(ftp, "vgg16-397923af.pth", "/home/huchuanwen/bishe/collaborativeDL/ftp/test/vgg16.pth") print(best_predicted_arm) log.logSend("INFO device " + localhost + " select model vgg16") zk.stop() N = p.size x = np.arange(N) label = ("VGG16","alexnet","resnet-18") plt.bar(x, p, width=0.5,label="models",tick_label=label) plt.legend() plt.show()
def selectModel(): # 配置信息 conf = configparser.ConfigParser() conf.read('../conf/inf.conf') kafkaHost = conf.get("ssd.kafkaServer", "hosts") localhost = conf.get('ssd.local', 'host') producer = KafkaProducer(bootstrap_servers=kafkaHost) # 连接kafka consoleChanel = conf.get("ssd.kafkaServer", "consoleChanel") zk = KazooClient(hosts="10.4.10.239:2181") zk.start() p = zk.get('/selectModel/param/p') # print(p) producer.send(consoleChanel, ("INFO device " + localhost + " get model selector parameters").encode()) p = pickle.loads(p[0]) best_predicted_arm = np.argmax(p) if best_predicted_arm == 0: ftp = FTPClient.ftpconnect("127.0.0.1", "ftp***", "Guest***") FTPClient.downloadfile( ftp, "vgg16-397923af.pth", "/home/huchuanwen/bishe/collaborativeDL/ftp/test/vgg16.pth") print(best_predicted_arm) producer.send(consoleChanel, ("INFO device " + localhost + " select model vgg16").encode()) zk.stop()