Feature transformers (VectorAssembler, StandardScaler) in Apache Spark

Introduction

Feature transformers help prepare data for machine learning by combining and scaling features. This makes models work better and faster.

When you have many columns and want to combine them into one feature vector.

When your features have different scales and you want to make them comparable.

Before training a machine learning model that expects input as a single vector.

When you want to improve model accuracy by normalizing data.

When you want to simplify feature handling in Spark pipelines.

Syntax

Apache Spark

from pyspark.ml.feature import VectorAssembler, StandardScaler

assembler = VectorAssembler(inputCols=["col1", "col2"], outputCol="features")
scaler = StandardScaler(inputCol="features", outputCol="scaledFeatures", withMean=True, withStd=True)

VectorAssembler combines multiple columns into one vector column.

StandardScaler scales features to have zero mean and unit variance.

Examples

Combine height and weight columns into a single features vector.

Apache Spark

assembler = VectorAssembler(inputCols=["height", "weight"], outputCol="features")

Scale the features vector to have mean 0 and standard deviation 1.

Apache Spark

scaler = StandardScaler(inputCol="features", outputCol="scaledFeatures", withMean=True, withStd=True)

Apply VectorAssembler then StandardScaler on a DataFrame.

Apache Spark

assembled_data = assembler.transform(df)
scaled_data = scaler.fit(assembled_data).transform(assembled_data)

Sample Program

This program creates a Spark DataFrame with height and weight columns. It uses VectorAssembler to combine these into a single vector column called 'features'. Then, it uses StandardScaler to scale these features so they have zero mean and unit variance, creating a new column 'scaledFeatures'. Finally, it shows the original and scaled features.

Apache Spark

from pyspark.sql import SparkSession
from pyspark.ml.feature import VectorAssembler, StandardScaler

# Start Spark session
spark = SparkSession.builder.appName("FeatureTransformersExample").getOrCreate()

# Sample data
data = [(1, 10.0, 100.0), (2, 20.0, 200.0), (3, 30.0, 300.0)]
columns = ["id", "height", "weight"]
df = spark.createDataFrame(data, columns)

# Combine height and weight into features vector
assembler = VectorAssembler(inputCols=["height", "weight"], outputCol="features")
assembled_df = assembler.transform(df)

# Scale features to zero mean and unit variance
scaler = StandardScaler(inputCol="features", outputCol="scaledFeatures", withMean=True, withStd=True)
scaler_model = scaler.fit(assembled_df)
scaled_df = scaler_model.transform(assembled_df)

# Show results
scaled_df.select("id", "features", "scaledFeatures").show(truncate=False)

spark.stop()

Important Notes

VectorAssembler output is a vector column, not separate columns.

StandardScaler requires fitting on data before transforming.

Scaling helps algorithms that are sensitive to feature scale, like k-means or logistic regression.

Summary

Use VectorAssembler to combine multiple columns into one vector column.

Use StandardScaler to scale features to zero mean and unit variance.

These transformers prepare data for better machine learning model performance.

Practice

1. What is the main purpose of the VectorAssembler in Apache Spark's feature transformers?

easy

A. To remove missing values from the dataset

B. To scale features to have zero mean and unit variance

C. To combine multiple columns into a single vector column

D. To split a vector column into multiple columns

Solution

1. Step 1: Understand VectorAssembler's role

   VectorAssembler is used to combine several feature columns into one vector column, which is needed for ML algorithms.

2. Step 2: Differentiate from other transformers

   StandardScaler scales features, but VectorAssembler only combines columns into a vector.

3. Final Answer:

   To combine multiple columns into a single vector column -> Option C

4. Quick Check:

   VectorAssembler = combine columns [OK]

Hint: VectorAssembler joins columns into one vector column [OK]

Common Mistakes:

• Confusing VectorAssembler with StandardScaler
• Thinking it splits vectors instead of combining
• Assuming it handles missing values

2. Which of the following is the correct way to create a VectorAssembler in PySpark to combine columns 'age' and 'income' into 'features'?

easy

A. VectorAssembler(inputCols=['age', 'income'], outputCol='features')

B. VectorAssembler(outputCols=['age', 'income'], inputCol='features')

C. VectorAssembler(inputCol=['age', 'income'], outputCols='features')

D. VectorAssembler(columns=['age', 'income'], output='features')

Solution

1. Step 1: Recall VectorAssembler syntax

   The correct parameters are inputCols (list of input columns) and outputCol (name of output vector column).

2. Step 2: Match parameters with options

   VectorAssembler(inputCols=['age', 'income'], outputCol='features') uses inputCols and outputCol correctly; others misuse parameter names or types.

3. Final Answer:

   VectorAssembler(inputCols=['age', 'income'], outputCol='features') -> Option A

4. Quick Check:

   inputCols and outputCol are correct parameters [OK]

Hint: Use inputCols and outputCol parameters exactly [OK]

Common Mistakes:

• Swapping inputCols and outputCol names
• Using singular inputCol instead of inputCols
• Using wrong parameter names like columns or output

3. Given the following PySpark code, what will be the output of scaled_features.show()?

from pyspark.ml.feature import VectorAssembler, StandardScaler
from pyspark.sql import SparkSession

spark = SparkSession.builder.getOrCreate()
data = [(1, 10.0, 100.0), (2, 20.0, 200.0), (3, 30.0, 300.0)]
df = spark.createDataFrame(data, ['id', 'feature1', 'feature2'])

assembler = VectorAssembler(inputCols=['feature1', 'feature2'], outputCol='features')
assembled = assembler.transform(df)

scaler = StandardScaler(inputCol='features', outputCol='scaledFeatures', withMean=True, withStd=True)
scalerModel = scaler.fit(assembled)
scaled_features = scalerModel.transform(assembled).select('id', 'scaledFeatures')

medium

A. A DataFrame with 'id' and 'scaledFeatures' columns where features are scaled to zero mean and unit variance

B. A DataFrame with 'id' and 'features' columns unchanged

C. An error because StandardScaler cannot be applied after VectorAssembler

D. A DataFrame with only 'scaledFeatures' column without 'id'

Solution

1. Step 1: Understand VectorAssembler output

   VectorAssembler creates a 'features' vector column combining 'feature1' and 'feature2'.

2. Step 2: Apply StandardScaler

   StandardScaler scales 'features' to zero mean and unit variance, outputting 'scaledFeatures'.

3. Final Answer:

   A DataFrame with 'id' and 'scaledFeatures' columns where features are scaled to zero mean and unit variance -> Option A

4. Quick Check:

   VectorAssembler + StandardScaler = scaled features [OK]

Hint: StandardScaler scales VectorAssembler output vector [OK]

Common Mistakes:

• Expecting original columns unchanged after scaling
• Thinking StandardScaler causes error after VectorAssembler
• Missing 'id' column in output

4. Identify the error in the following PySpark code snippet using StandardScaler:

assembler = VectorAssembler(inputCols=['f1', 'f2'], outputCol='features')
df2 = assembler.transform(df)
scaler = StandardScaler(inputCol='features', outputCol='scaled', withMean=False, withStd=True)
scaled_df = scaler.transform(df2)

medium

A. Wrong inputCol name in StandardScaler

B. Missing call to fit() before transform() on StandardScaler

C. VectorAssembler outputCol should be 'scaled' not 'features'

D. withMean must be True for StandardScaler to work

Solution

1. Step 1: Check StandardScaler usage

   StandardScaler requires calling fit() on data before transform().

2. Step 2: Identify missing fit() call

   The code calls transform() directly on scaler without fit(), causing error.

3. Final Answer:

   Missing call to fit() before transform() on StandardScaler -> Option B

4. Quick Check:

   StandardScaler needs fit() before transform() [OK]

Hint: Always fit StandardScaler before transform [OK]

Common Mistakes:

• Calling transform() without fit() on StandardScaler
• Confusing inputCol and outputCol names
• Assuming withMean must be True

5. You have a dataset with columns 'height', 'weight', and 'age'. You want to prepare features for a machine learning model by combining these columns into a single vector and scaling them to zero mean and unit variance. Which sequence of steps correctly achieves this in PySpark?

hard

A. Use StandardScaler first on raw columns, then combine with VectorAssembler

B. Apply StandardScaler separately on each column, then combine with VectorAssembler

C. Combine columns with VectorAssembler and use StandardScaler without fitting

D. Use VectorAssembler to combine columns, then fit and transform with StandardScaler on the assembled vector

Solution

1. Step 1: Combine columns using VectorAssembler

   VectorAssembler merges 'height', 'weight', and 'age' into one vector column for ML models.

2. Step 2: Fit and transform the assembled vector with StandardScaler

   StandardScaler requires fitting on the vector column to scale features properly.

3. Final Answer:

   Use VectorAssembler to combine columns, then fit and transform with StandardScaler on the assembled vector -> Option D

4. Quick Check:

   Combine first, then scale with fit() [OK]

Hint: Combine columns first, then fit StandardScaler [OK]

Common Mistakes:

• Scaling columns before combining
• Skipping fit() on StandardScaler
• Combining after scaling