Skip to content

added probability training and prediction#5

Merged
zeyiwen merged 1 commit intoXtra-Computing:masterfrom
shijiashuai:master
Nov 5, 2016
Merged

added probability training and prediction#5
zeyiwen merged 1 commit intoXtra-Computing:masterfrom
shijiashuai:master

Conversation

@shijiashuai
Copy link
Copy Markdown
Collaborator

probability training and prediction

to train with probabilty
set param.probabilty = 1
to predict with probability
y_predict = model.predict(x_test,true)
to get probability directly
prob = model.predictProbability(x_test)

example

see trainingFunction.cu:48

@zeyiwen zeyiwen merged commit d2d9f3e into Xtra-Computing:master Nov 5, 2016
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants