pip install pydeep python import pydeep dataset = pydeep.Dataset('path/to/dataset') model = pydeep.Model() model.add(pydeep.layers.Conv2D(64, (3,3), activation='relu', input_shape=(32,32,3))) model.add(pydeep.layers.MaxPooling2D((2,2))) model.add(pydeep.layers.Conv2D(128, (3,3), activation='relu')) model.add(pydeep.layers.MaxPooling2D((2,2))) model.add(pydeep.layers.Flatten()) model.add(pydeep.layers.Dense(64, activation='relu')) model.add(pydeep.layers.Dense(10, activation='softmax')) model.compile(optimizer='sgd', loss='categorical_crossentropy', metrics=['accuracy']) model.fit(dataset.train_images, dataset.train_labels, epochs=10, batch_size=32) loss, accuracy = model.evaluate(dataset.test_images, dataset.test_labels) print('Test loss:', loss) print('Test accuracy:', accuracy)


上一篇:
下一篇:
切换中文