fix multi class classification

[masoud-khalilian]

original image -> when fetching -> shape: , type: -> when giving it to the model -> shape: , type:

label -> when fetching -> shape: , type: -> after encoding -> shape: , type: -> when using as metric for train -> shape: , type: -> when using as finding the accuracy metric -> shape: , type:

output -> when getting feeded the image -> shape: , type: -> when try extract the final result -> shape: , type: