码迷,mamicode.com
首页 > 其他好文 > 详细

tensorflow2中pydot问题

时间:2020-12-21 12:05:44      阅读:0      评论:0      收藏:0      [点我收藏+]

标签:cti   article   call   mode   图片   pat   嵌入   features   tps   

1. pydot` failed to call GraphViz.

 首先去下载这个 。https://blog.csdn.net/shangxiaqiusuo1/article/details/85283432(下载包这个博客里面有飞机票)

2。参照上面那个博客的顺序去把GraphViz,pydot,pydot-ng下载完。

3.  系统变量的添加,参照这个博客https://blog.csdn.net/hjxinkkl/article/details/89483033中的

  import os

  os.environ["PATH"] += ";D:/Program/Graphviz2.38/bin/

4. 

num_words = 2000
num_tags = 12
num_departments = 4

# 输入
body_input = keras.Input(shape=(None,), name=‘body‘)
title_input = keras.Input(shape=(None,), name=‘title‘)
tag_input = keras.Input(shape=(num_tags,), name=‘tag‘)

# 嵌入层
body_feat = layers.Embedding(num_words, 64)(body_input)
title_feat = layers.Embedding(num_words, 64)(title_input)

# 特征提取层
body_feat = layers.LSTM(32)(body_feat)
title_feat = layers.LSTM(128)(title_feat)
features = layers.concatenate([title_feat,body_feat, tag_input])

# 分类层
priority_pred = layers.Dense(1, activation=‘sigmoid‘, name=‘priority‘)(features)
department_pred = layers.Dense(num_departments, activation=‘softmax‘, name=‘department‘)(features)


# 构建模型
model = keras.Model(inputs=[body_input, title_input, tag_input],
outputs=[priority_pred, department_pred])
model.summary()
keras.utils.plot_model(model, ‘multi_model.png‘, show_shapes=True)

最后一个行就可以完成图片的绘制

tensorflow2中pydot问题

标签:cti   article   call   mode   图片   pat   嵌入   features   tps   

原文地址:https://www.cnblogs.com/dhName/p/14145810.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!