prepared version 1.2.4

alexware69 · Apr 28, 2022 · 06f2280 · 06f2280
1 parent b5c1d37
commit 06f2280
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 17 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/src/Cid3.kt b/src/Cid3.kt
@@ -13,7 +13,7 @@ import kotlin.system.exitProcess
 
 
 class Cid3 : Serializable {
-    private val version = "1.2.3"
+    private val version = "1.2.4"
     private var createdWith = ""
     enum class AttributeType {
         Discrete, Continuous, Ignore
@@ -1595,7 +1595,7 @@ class Cid3 : Serializable {
         }
 
         //If an attribute name is passed just do that attribute
-        if (attributeName.isNotEmpty()){
+        if (attributeName.isNotEmpty() && attributeName != "Class" && attributeNames.contains(attributeName)){
             var selectedAtt: Triple<Int,Double,Double> = Triple(0,0.0,0.0)
             for (att in attributeImportance){
                 if (attributeNames[att.first] == attributeName) {
@@ -1605,6 +1605,10 @@ class Cid3 : Serializable {
             }
             attributeImportance.clear()
             attributeImportance.add(selectedAtt)
+        } else {
+            print("\n")
+            print("Error: Attribute name not found.")
+            exitProcess(1)
         }
 
         val sortedList: List<Triple<Int, Double, Double>> = if (criteria == Criteria.Certainty)