spam_010.swift

func countVector(sentence:String) -> [Int:Int]? {
    var vec = [Int:Int]()
    for word in self.tokenize(sentence) {
        if let pos = self.vocabulary[word] {
            if let i = vec[pos] {
                vec[pos] = i+1
            } else {
                vec[pos] = 1
            }
        }
    }
    return vec
}