forked from Open-CT/openbrain
25 lines
582 B
Go
25 lines
582 B
Go
package object
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
)
|
|
|
|
func TestDoVectorsetTsne(t *testing.T) {
|
|
InitConfig()
|
|
|
|
dimension := 50
|
|
|
|
//vectorset := getVectorset("admin", "百度百科")
|
|
vectorset := getVectorset("admin", "wordVector_utf-8")
|
|
vectorset.LoadVectors("../../tmpFiles/")
|
|
vectorset.DoTsne(dimension)
|
|
|
|
vectorset.Name = fmt.Sprintf("%s_Dim_%d", vectorset.Name, dimension)
|
|
vectorset.FileName = fmt.Sprintf("%s_Dim_%d.csv", vectorset.FileName, dimension)
|
|
vectorset.FileSize = ""
|
|
vectorset.Dimension = dimension
|
|
vectorset.WriteVectors("../../tmpFiles/")
|
|
AddVectorset(vectorset)
|
|
}
|