brucechou1983/CheXNet-Keras

vec should not be hard-code intialized to length of 14

taoyilee opened this issue · 2 comments

taoyilee commented

CheXNet-Keras/generator.py

Line 41 in f01a206

vec = np.zeros(14)

suggestion: use len(class_names) to initialize vec instead

taoyilee commented

Pull request: #3

taoyilee commented

Same thing here

CheXNet-Keras/callback.py

Line 11 in f01a206

def load_generator_data(generator, steps, class_num=14):