Commit 7e36a9c6 authored by Gabriel Kirsten's avatar Gabriel Kirsten
Browse files

removed class name atribuition
parent 25f5fac4
......@@ -272,9 +272,6 @@ class CNNPseudoLabel(Classifier):
one_hot_output=one_hot_output.tolist()
for index in range(0, len(one_hot_output)):
one_hot_output[index]=CLASS_NAMES[one_hot_output[index]]
return one_hot_output
def train(self, dataset, training_data, force=False):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment