体裁检测AI(架构)

时间:2019-12-23 22:20:25

标签: python tensorflow keras artificial-intelligence

我的神经网络目前还没有真正学习。我目前正在this视频上使用该模型。


结构

def get_conv_model():
    model = Sequential()
    model.add(Conv2D(16, (3, 3), activation='relu', strides=(1, 1), padding='same',     input_shape=input_shape))
    model.add(Conv2D(32, (3, 3), activation='relu', strides=(1, 1), padding='same'))
    model.add(Conv2D(64, (3, 3), activation='relu', strides=(1, 1), padding='same'))
    model.add(Conv2D(128, (3, 3), activation='relu', strides=(1, 1), padding='same'))
    model.add(MaxPool2D((2, 2)))
    model.add(Dropout(0.5))
    model.add(Flatten())
    model.add(Dense(128, activation='relu'))
    model.add(Dense(64, activation='relu'))
    # model.add(Dense(32, activation='relu'))
    # model.add(Dense(10, activation='relu'))
    model.add(Dense(config.cat, activation='softmax'))
    model.summary()
    model.compile(loss='categorical_crossentropy',
                  optimizer='adam', metrics=['acc'])
    return model

数据集

  • 3种流派* 170首曲调(共510首歌曲)
  • 每首歌曲随机剪切1秒

过程

  • 20个时代

结果

  • 约0.9214 acc(亏损0.2074)

但是,所有预测均仅基于3种类型之一。所以几乎50%的分类是错误的。

0 个答案:

没有答案