I aprecitate your contribution, i want to know which part shuld to change when i want to get binary classification, i just change 32 to 2 in model.py, after run train.py, I get wrong result when run predict.py, the prediction is very bad, can you give me some suggestion.
@kirilcvetkov92
I aprecitate your contribution, i want to know which part shuld to change when i want to get binary classification, i just change 32 to 2 in model.py, after run train.py, I get wrong result when run predict.py, the prediction is very bad, can you give me some suggestion.
@kirilcvetkov92