我從互聯網上的一些教程中使用 Keras 創建了聲音分類器構建。這是我的模型代碼
import tensorflow as tf
from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, InputLayer, Dropout, Conv1D, Conv2D, Flatten, Reshape, MaxPooling1D, MaxPooling2D, BatchNormalization, TimeDistributed
from tensorflow.keras.optimizers import Adam
model = Sequential()
model.add(Reshape((int(input_length / 40), 40), input_shape=(input_length, )))
model.add(Conv1D(8, kernel_size=3, activation='relu', padding='same'))
model.add(MaxPooling1D(pool_size=2, strides=2, padding='same'))
model.add(Dropout(0.25))
model.add(Conv1D(16, kernel_size=3, activation='relu', padding='same'))
model.add(MaxPooling1D(pool_size=2, strides=2, padding='same'))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(classes, activation='softmax', name='y_pred'))
opt = Adam(lr=0.005, beta_1=0.9, beta_2=0.999)
# this controls the batch size, or you can manipulate the tf.data.Dataset objects yourself
BATCH_SIZE = 32
train_dataset = train_dataset.batch(BATCH_SIZE, drop_remainder=False)
validation_dataset = validation_dataset.batch(BATCH_SIZE, drop_remainder=False)
model.compile(loss='categorical_crossentropy', optimizer=opt, metrics=['accuracy'])
model.fit(train_dataset, epochs=1000, validation_data=validation_dataset, verbose=2, callbacks=callbacks)
我的老師問我用于分類的演算法是什么(他說的是 K-NN、樸素貝葉斯、SVM 之類的東西),我不知道我在使用什么。
uj5u.com熱心網友回復:
您正在使用卷積神經網路 (CNN)
轉載請註明出處,本文鏈接:https://www.uj5u.com/net/375427.html
上一篇:TensorFlowTextVectorization從pickle加載后生成不帶填充的RaggedTensor
下一篇:ValueError:`logits`和`labels`必須具有相同的形狀,接收((None,10)vs(None,11,2,2))
