keras

作者: 一川烟草i蓑衣 | 来源:发表于2020-08-05 08:27 被阅读0次

    Keras设计了俩种构建模型的方式函数式模型API和顺序式模型API

    顺序式模型API构建模型示例:

    from keras.models import Sequential   #从models导入Sequential

    model = Sequential()

    from keras.layers import Dense

    model.add(Dense(units=64, activation='relu', input_dim=100))

    model.add(Dense(units=10, activation='softmax'))

    model.compile(optimizer='rmsprop',loss='categorical_crossentropy',metrics=['accuracy'])

    model.fit(data,labels, batch_size=32, nb_epoch=10, verbose=1)

    主要配置包括函数式模型API构建模型示例:

    from keras.layers import Input, Dense

    from keras.models import Model

    # 这部分返回一个张量

    inputs = Input(shape=(784,))

    # 层的实例是可调用的,它以张量为参数,并且返回一个张量

    x = Dense(64, activation='relu')(inputs)

    x = Dense(64, activation='relu')(x)

    predictions = Dense(10, activation='softmax')(x)

    # 这部分创建了一个包含输入层和三个全连接层的模型

    model = Model(inputs=inputs, outputs=predictions)

    model.compile(optimizer='rmsprop',

    loss='categorical_crossentropy',

    metrics=['accuracy'])

    model.fit(data, labels)# 开始训练

    模型编译参数设置

    主要配置包括:

    loss:损失函数–MSE、KL散度、交叉熵、对数损失函数等;源码:losses.py

    optimizer:优化函数–常用SGD、RMSprop、Adam等;源码:optimizers.py

    metrics:模型评估方法–准确率、AUC、F-score、包括损失函数代表的指标;源码:metrics.py

    模型训练参数设置

    相关文章

      网友评论

          本文标题:keras

          本文链接:https://www.haomeiwen.com/subject/uyigrktx.html