Skip to content

Commit b670adc

Browse files
author
Olcay Taner YILDIZ
committed
Added predictProbability method.
1 parent c12eed6 commit b670adc

2 files changed

Lines changed: 3 additions & 1 deletion

File tree

Classification/Model/NeuralNetworkModel.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,8 @@ def predict(self, instance: Instance) -> str:
227227
return self.classLabels[self.y.maxIndex()]
228228

229229
def predictProbability(self, instance: Instance) -> dict:
230+
self.createInputVector(instance)
231+
self.calculateOutput()
230232
result = {}
231233
for i in range(len(self.classLabels)):
232234
result[self.classLabels[i]] = self.y.getValue(i)

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
setup(
44
name='NlpToolkit-Classification',
5-
version='1.0.8',
5+
version='1.0.9',
66
packages=['Classification', 'Classification.Model', 'Classification.Model.DecisionTree', 'Classification.Filter',
77
'Classification.DataSet', 'Classification.Instance', 'Classification.Attribute',
88
'Classification.Parameter', 'Classification.Classifier', 'Classification.Experiment',

0 commit comments

Comments
 (0)