Keras-多输入多输出实例(多任务)

1、模型结果设计

a tuple (inputs, targets, sample_weights).

Keras设计多输出（多任务）使用fit_generator的步骤如下：

根据官方文档，定义一个generator或者一个class继承Sequence

class Batch_generator(Sequence):
 """
 用于产生batch_1, batch_2(记住是numpy.array格式转换)
 """
 y_batch = {'main':batch_1,'auxiliary':batch_2}
 return X_batch, y_batch

# or in another way
def batch_generator():
 """
 用于产生batch_1, batch_2(记住是numpy.array格式转换)
 """
 yield X_batch, {'main': batch_1,'auxiliary':batch_2}

重要的事情说三遍(亲自采坑，搜了一大圈才发现滴)：

如果是多输出（多任务）的时候，这里的target是字典类型

如果是多输出（多任务）的时候，这里的target是字典类型

如果是多输出（多任务）的时候，这里的target是字典类型

以上这篇Keras-多输入多输出实例(多任务)就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持来客网。