keras 入门之常用代码

1、指定tensorflow或者其他框架，使用哪个显卡运行代码

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "1"

如果想要禁止keras，或者tensorflow使用GPU，可以将这个值设为-1

 

2、访问import 其他文件夹下的py文件

在py文件所在的路径下建立文件   __init__.py  （可以是空文件）

import sys

sys.path.append(py文件所在的path)

 

3、显示keras 模型的结构

from keras.utils import plot_model
plot_model(model, to_file='model.png')

遇到无法工作的问题，按照提示下载文件，安装后将bin目录加入到path里就行了。

 

4、为了保证实验的可重现性，需要制定随机化种子

seed=1023

random.seed(seed)
numpy.random.seed(seed)

如果用的是torch，还要再加两个

torch.cuda.manual_seed(seed) 
torch.manual_seed(seed)

 

5、输出一个tensorflow的内容

with tf.Session() as sess:

    print(张量名.eval())