Изменения

Перейти к: навигация, поиск

Примеры кода на Scala

136 байт добавлено, 22 январь
Рекуррентные нейронные сети
Пример кода, с использованием билиотеки DeepLearning.scala
<span style="color:#3D9970>// Задание слоёв</span>
'''def '''tanh(x: INDArrayLayer): INDArrayLayer = {
'''val '''exp_x = hyperparameters.exp(x)
}
<span style="color:#3D9970>// Определение структуры</span>
'''val '''batches = data.zip(data.tail).grouped(seqLength).toVector
type WithHiddenLayer[A] = (A, INDArrayLayer)
}
<span style="color:#3D9970>// Определение одного шага обучения</span>
'''def '''initH = INDArrayLayer(Nd4j.zeros(hiddenSize, 1).forward)
'''def '''singleRound(initprevloss: Losses): Future[Losses] =
}
<span style="color:#3D9970>// Обучение сети</span>
'''def '''unsafePerformFuture[A](f: Future[A]): A = Await.result(f.toScalaFuture, Duration.Inf)
'''val '''losses = unsafePerformFuture(allRounds)
Анонимный участник

Навигация