eeva/tensorflow-haskell
1
0
Fork 0
mirror of https://github.com/tensorflow/haskell.git synced 2024-11-26 21:09:44 +01:00
Code Issues Releases Wiki Activity

Fixed CI and added indicator. (#15)

Browse source 
* Sorted test names.

* Don't require a terminal to run tests.

Should resolve "the input device is not a TTY" problem with Jenkins.

* Added build indicator to README.md.

* Fixed up URL.
This commit is contained in:
Greg Steuck 2016-10-28 18:08:32 -07:00 committed by fkm3
parent 28bfe005da
commit 9f2f4e2877
3 changed files with 9 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
README.md
View file

@ -1,3 +1,5 @@
[![Build Status](https://ci.tensorflow.org/buildStatus/icon?job=tensorflow-haskell-master)](https://ci.tensorflow.org/job/tensorflow-haskell-master)
The tensorflow-haskell package provides Haskell bindings to The tensorflow-haskell package provides Haskell bindings to
[TensorFlow](https://www.tensorflow.org/). [TensorFlow](https://www.tensorflow.org/).

2
ci_build/outer_launch_tests.sh
View file

@ -8,4 +8,4 @@ IMAGE_NAME=tensorflow/haskell/ci_build:v0
git submodule update git submodule update
docker build -t $IMAGE_NAME -f ci_build/Dockerfile . docker build -t $IMAGE_NAME -f ci_build/Dockerfile .
docker run -ti $IMAGE_NAME stack test docker run $IMAGE_NAME stack test

12
tensorflow-nn/tests/NNTest.hs
View file

@ -78,20 +78,20 @@ testLogisticOutput = testCase "testLogisticOutput" $ do
vTargets = TF.vector $ targets inputs vTargets = TF.vector $ targets inputs
tfLoss = TF.sigmoidCrossEntropyWithLogits vLogits vTargets tfLoss = TF.sigmoidCrossEntropyWithLogits vLogits vTargets
ourLoss = V.fromList $ sigmoidXentWithLogits (logits inputs) (targets inputs) ourLoss = V.fromList $ sigmoidXentWithLogits (logits inputs) (targets inputs)
r <- run tfLoss r <- run tfLoss
assertAllClose r ourLoss assertAllClose r ourLoss
testLogisticOutputMultipleDim = testLogisticOutputMultipleDim =
testCase "testLogisticOutputMultipleDim" $ do testCase "testLogisticOutputMultipleDim" $ do
let inputs = defInputs let inputs = defInputs
shape = [2, 2, 2] shape = [2, 2, 2]
vLogits = TF.constant shape (logits inputs) vLogits = TF.constant shape (logits inputs)
vTargets = TF.constant shape (targets inputs) vTargets = TF.constant shape (targets inputs)
tfLoss = TF.sigmoidCrossEntropyWithLogits vLogits vTargets tfLoss = TF.sigmoidCrossEntropyWithLogits vLogits vTargets
ourLoss = V.fromList $ sigmoidXentWithLogits (logits inputs) (targets inputs) ourLoss = V.fromList $ sigmoidXentWithLogits (logits inputs) (targets inputs)
r <- run tfLoss r <- run tfLoss
assertAllClose r ourLoss assertAllClose r ourLoss
@ -101,7 +101,7 @@ testGradientAtZero = testCase "testGradientAtZero" $ do
vLogits = TF.vector $ logits inputs vLogits = TF.vector $ logits inputs
vTargets = TF.vector $ targets inputs vTargets = TF.vector $ targets inputs
tfLoss = TF.sigmoidCrossEntropyWithLogits vLogits vTargets tfLoss = TF.sigmoidCrossEntropyWithLogits vLogits vTargets
r <- run $ do r <- run $ do
l <- tfLoss l <- tfLoss
TF.gradients l [vLogits] TF.gradients l [vLogits]
@ -113,7 +113,7 @@ run = TF.runSession . TF.buildAnd TF.run
main :: IO () main :: IO ()
main = googleTest [ testLogisticOutput main = googleTest [ testGradientAtZero
, testLogisticOutput
, testLogisticOutputMultipleDim , testLogisticOutputMultipleDim
, testGradientAtZero
] ]