Fixed: pharo-ai#3 Copied method from DataFrame

jodz4k · Apr 23, 2024 · 816db25 · 816db25
1 parent 162743f
commit 816db25
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/AI-DataPreProcessing-Encoders/DataFrame.extension.st b/src/AI-DataPreProcessing-Encoders/DataFrame.extension.st
@@ -0,0 +1,17 @@
+Extension { #name : #DataFrame }
+
+{ #category : #'*AI-DataPreProcessing-Encoders' }
+DataFrame >> dataPreProcessingEncodeWith: anEncoder [
+	"This method is here to speed up pharo-ai/data-preprocessing algos without coupling both projects."
+
+	| copy cache |
+	copy := self copy.
+	cache := IdentityDictionary new.
+	self columns doWithIndex: [ :dataSerie :columnIndex |
+		| category |
+		category := cache at: columnIndex ifAbsentPut: [ ((anEncoder categories at: columnIndex) collectWithIndex: [ :elem :index | elem -> index ]) asDictionary ].
+		dataSerie doWithIndex: [ :element :rowIndex |
+			copy at: rowIndex at: columnIndex put: (category at: element ifAbsent: [ AIMissingCategory signalFor: element ]) ] ].
+
+	^ copy
+]