From 17f8341688b9faa0d2771880d1936ef44e6fc931 Mon Sep 17 00:00:00 2001 From: TRD Date: Thu, 7 Oct 2021 18:14:16 +0530 Subject: [PATCH] minor fix --- blobcity/utils/ProblemType.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/blobcity/utils/ProblemType.py b/blobcity/utils/ProblemType.py index 8fe47ca..1c36053 100644 --- a/blobcity/utils/ProblemType.py +++ b/blobcity/utils/ProblemType.py @@ -38,7 +38,7 @@ def checkType(self,data): if(data.dtype in ['object']): return dict({'type':'Classification'}) else: target_length =len(np.unique(data)) - if data.dtype in ['int','float'] and target_length<=50: + if data.dtype in ['int64','float64','int32','float32','int16','float16'] and target_length<=10: return dict({'type':'Classification'}) else: return dict({'type':'Regression'})