relu: run notmnist using relu activation and draw the chart
[wip] word2vec: work in progress implementation of word2vec
This commit is contained in:
@ -43,7 +43,7 @@ module Main where
|
||||
|
||||
let session = def { learningRate = 0.5
|
||||
, batchSize = 32
|
||||
, epochs = 10
|
||||
, epochs = 9
|
||||
, network = rnetwork
|
||||
, training = zip trinputs trlabels
|
||||
, test = zip teinputs telabels
|
||||
|
34
examples/word2vec.hs
Normal file
34
examples/word2vec.hs
Normal file
@ -0,0 +1,34 @@
|
||||
{-# LANGUAGE RecordWildCards #-}
|
||||
{-# LANGUAGE FlexibleContexts #-}
|
||||
{-# LANGUAGE ScopedTypeVariables #-}
|
||||
|
||||
module Main where
|
||||
import Sibe
|
||||
import Sibe.Word2Vec
|
||||
import Sibe.Utils
|
||||
import Data.Default.Class
|
||||
import qualified Data.Vector.Storable as V
|
||||
import Data.List (sortBy)
|
||||
import Data.Function (on)
|
||||
import Numeric.LinearAlgebra
|
||||
import System.IO
|
||||
import Data.List.Split
|
||||
|
||||
main = do
|
||||
{-ds <- do
|
||||
content <- readFile "examples/doc-classifier-data/data-reuters"
|
||||
let splitted = splitOn (replicate 10 '-' ++ "\n") content
|
||||
d = concatMap (tail . lines) (take 100 splitted)
|
||||
return d-}
|
||||
let ds = ["I like deep learning", "I like NLP", "I enjoy flying"]
|
||||
|
||||
let session = def { learningRate = 0.8
|
||||
, batchSize = 10
|
||||
, epochs = 1000
|
||||
} :: Session
|
||||
w2v = def { docs = ds }:: Word2Vec
|
||||
|
||||
|
||||
r <- word2vec w2v session
|
||||
{-print r-}
|
||||
return ()
|
Reference in New Issue
Block a user