Устанавливайте единицы для слоя динамически в соответствии с набором данных

Я пытаюсь изменить размер единицы для последнего слоя в соответствии с набором данных. Это абстракция моего кода, но она не работает.

class cnn_model:
 num_classes = 1

 model.add(layers.Conv2D(128, (3, 3), activation='relu'))
 model.add(layers.MaxPooling2D((2, 2)))
 model.add(layers.Conv2D(256, (3, 3), activation='relu'))
 model.add(layers.MaxPooling2D((2, 2)))
 model.add(layers.Dense(num_classes, activation='softmax'))


@staticmethod
def train_two():
 cnn_mod = cnn_model
 cnn_mod.num_classes = 2
 model = cnn_mod.model

@staticmethod
def train_three():
 cnn_mod = cnn_model
 cnn_mod.num_classes = 3
 model = cnn_mod.model

python tensorflow keras

07.04.2019 14:50

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Просто передайте количество классов в качестве аргумента при создании экземпляра нового класса CNNModel:

import tensorflow as tf
from tensorflow.keras import layers

class CNNModel:

    def __init__(self, num_classes=2):
        self.num_classes = num_classes
        self.model = tf.keras.models.Sequential()
        self.model.add(layers.Conv2D(128, (3, 3), activation='relu'))
        self.model.add(layers.MaxPooling2D((2, 2)))
        self.model.add(layers.Conv2D(256, (3, 3), activation='relu'))
        self.model.add(layers.MaxPooling2D((2, 2)))
        self.model.add(layers.Dense(self.num_classes, activation='softmax'))

cnnmodel = CNNModel(num_classes=3)
kerasmodel = cnnmodel.model
print(cnnmodel.num_classes) # 3

Я бы также посоветовал вам прочитать Соглашения об именах и Отступ в PEP8.

07.04.2019 14:55

Другие вопросы по теме

TensorFlow: вызов графа внутри другого графа

Вложенный тензорный поток map_fn объединяет два тензора

TensorFlow: нарежьте Tensor и сохраните исходную форму

Альтернативный способ решить «Uncaught (в обещании) TypeError: fs.writeFile не является функцией» для js tensorflow, работающего в chrome

Алгоритм DQN не сходится на CartPole-v0

Веса Tensorflow не восстанавливаются даже после восстановления из файла ckpt

Как использовать tf.estimator.DNNClassifier (Scikit Flow?)

Нулевая точность Tensorflow CNN?

Keras: исчезновение слоя или изменение переменной в сеансе

Многозадачное обучение с использованием оценок в качестве признаков

Устанавливайте единицы для слоя динамически в соответствии с набором данных

Ответы 1

Другие вопросы по теме

Похожие вопросы