[ad8447]: / (2) PyTorch_HistoTNet / util / getClassCount.py

Download this file

18 lines (12 with data), 324 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
import numpy as np
from util import pause
def getClassCount(classVec):
#print(classVec[0])
classCounts = np.zeros(len(classVec[0]))
for vec in classVec:
classCounts += vec
#print(vec)
#print('Uscita dalla funzione classcount')
#print(classCounts)
#pause()
return classCounts