Блог

Когда я запускаю net.fit(X,y) in skorch , он выводит все полезные данные — есть ли способ изменить функцию проверки, например, на AUC, F1 или какую-то третью функцию вместо стандартной точности? Или, если это невозможно, есть ли способ извлечь набор проверки из сети и вычислить его «вручную»?

Вы можете использовать любую метрику sklearn, которую хотите, в сочетании с обратным вызовом EpochScoring, таким образом, AUC и F1 работают просто отлично. Взгляните на этот блокнот, чтобы увидеть, как это работает на практике. Вы даже можете реализовать свою собственную метрику, если она соответствует API sklearn.