VectorIndexer pyspark

from pyspark.ml import Pipeline
from pyspark.ml.regression import LinearRegression
from pyspark.ml.feature import VectorIndexer
from pyspark.ml.evaluation import RegressionEvaluator

# Automatically identify categorical features, and index them.
# We specify maxCategories so features with > 4 distinct values are treated as continuous.

featureIndexer = VectorIndexer(inputCol="features", \
                               outputCol="indexedFeatures",\
                               maxCategories=4).fit(transformed)

data = featureIndexer.transform(transformed)

Source: runawayhorse001.github.io

Similar codes

4 votes177 views2 answers
Python get index of substring in liast
Python
15 votes148 views4 answers
How to stop python for some time in python
Python
2 votes182 views1 answer
Ski learn decision tree
Python
0 vote160 views1 answer
Remove first 2 rows in pandas
Python
0 vote172 views1 answer
Apply same shuffle to two arrays numpy
Python
0 vote188 views1 answer
Gauss elimination python numpy
Python
-2 votes165 views4 answers
DFS
Python
15 votes183 views6 answers
Matplotlib.pyplot
Python
2 votes173 views1 answer
Python dictonary of dictonary
Python
1 vote156 views1 answer
Convert pandas dataframe to numpy dataframe
Python