[Keras] Keras使用中遇到的一些坑~

主要内容

如题,遇到的时候会黑人问号脸的Keras Issues


1. Class Weights问题

问题描述:Keras在多分类或多二分类时使用Class Weights时会报错,提示class not exists,这是一个已知的Keras Bug,有一个work around的方法是,在label vector前面加两个位置,分别赋值0和1,举个例子:

# Original label vector
y_ori = [1, 0, 0, 0]

# Work Around label vector for Keras
y_ = [0, 1, 1, 0, 0, 0]

2. Multi-GPU Compile后遇到的问题

第一:神TM在Compile 到Multi-GPU之后,保存模型时使用save函数就报错了,需要使用save_weight

第二:使用Multi-GPU Compile 的模型在predict的时候,batch的size必须是GPU number的倍数…

你可能感兴趣的:(机器学习,自然语言处理)