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()
