Add syntatic suggar

Author: jecisc

src/AI-DataImputers-Tests/AISimpleImputerTest.class.st

@@ -90,39 +90,35 @@ AISimpleImputerTest >> testTransform1DColl [
 { #category : #tests }
 AISimpleImputerTest >> testTransformAverage [
 
-	imputer
-		useAverage;
-		fit: collection.
+	imputer := AISimpleImputer average.
+	imputer fit: collection.
 
 	self assert: (imputer transform: collection) equals: #( #( 7 2 5 6 ) #( 7 2 5 9 ) #( 10 2 5 6 ) )
 ]
 
 { #category : #tests }
 AISimpleImputerTest >> testTransformConstant [
 
-	imputer
-		useConstant: 10;
-		fit: collection.
+	imputer := AISimpleImputer constant: 10.
+	imputer fit: collection.
 
-	self assert: (imputer transform: collection) equals: #( #( 7 2 5 6 ) #( 7 10 5 9 ) #( 10 2 10 6 ) ).
+	self assert: (imputer transform: collection) equals: #( #( 7 2 5 6 ) #( 7 10 5 9 ) #( 10 2 10 6 ) )
 ]
 
 { #category : #tests }
 AISimpleImputerTest >> testTransformMedian [
 
-	imputer
-		useAverage;
-		fit: collection.
+	imputer := AISimpleImputer median.
+	imputer fit: collection.
 
 	self assert: (imputer transform: collection) equals: #( #( 7 2 5 6 ) #( 7 2 5 9 ) #( 10 2 5 6 ) )
 ]
 
 { #category : #tests }
 AISimpleImputerTest >> testTransformMostFrequent [
 
-	imputer
-		useMostFrequent;
-		fit: collection.
+	imputer := AISimpleImputer mostFrequent.
+	imputer fit: collection.
 
 	self assert: (imputer transform: collection) equals: #( #( 7 2 5 6 ) #( 7 2 5 9 ) #( 10 2 5 6 ) )
 ]

src/AI-DataImputers/AISimpleImputer.class.st

@@ -72,6 +72,38 @@ Class {
 	#category : #'AI-DataImputers'
 }
 
+{ #category : #'instance creation' }
+AISimpleImputer class >> average [
+
+	^ self new
+		  useAverage;
+		  yourself
+]
+
+{ #category : #'instance creation' }
+AISimpleImputer class >> constant: aNumber [
+
+	^ self new
+		  useConstant: aNumber;
+		  yourself
+]
+
+{ #category : #'instance creation' }
+AISimpleImputer class >> median [
+
+	^ self new
+		  useMedian;
+		  yourself
+]
+
+{ #category : #'instance creation' }
+AISimpleImputer class >> mostFrequent [
+
+	^ self new
+		  useMostFrequent;
+		  yourself
+]
+
 { #category : #private }
 AISimpleImputer >> ensureIs2D: aCollection [