From cdd77b1dfa0c025dd2b66b4cfd83ca208003597f Mon Sep 17 00:00:00 2001
From: Jamie Mac <macster110@gmail.com>
Date: Fri, 20 Jan 2023 11:47:02 +0000
Subject: [PATCH] Bug fix deep learning classifier

The number of class names was not being read from import file- fixed
---
 .../dlClassification/genericModel/GenericModelParser.java       | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/rawDeepLearningClassifier/dlClassification/genericModel/GenericModelParser.java b/src/rawDeepLearningClassifier/dlClassification/genericModel/GenericModelParser.java
index bf7893bb..3c562aac 100644
--- a/src/rawDeepLearningClassifier/dlClassification/genericModel/GenericModelParser.java
+++ b/src/rawDeepLearningClassifier/dlClassification/genericModel/GenericModelParser.java
@@ -283,6 +283,8 @@ public class GenericModelParser {
 			classNames[i]=classNames[i].trim(); //remove whitespace
 
 		}
+		
+		paramsClone.numClasses = classNames.length;
 
 		
 		if (classNameManager!=null) {