Description

Map index to string.

Parameters

Name Description Type Required？ Default Value
modelName Name of the model String ✓
selectedCol Name of the selected column used for processing String ✓
reservedCols Names of the columns to be retained in the output table String[] null
outputCol Name of the output column String null

Script Example

Code

  1. data = np.array([
  2.     ["football"],
  3.     ["football"],
  4.     ["football"],
  5.     ["basketball"],
  6.     ["basketball"],
  7.     ["tennis"],
  8. ])
  10. df_data = pd.DataFrame({
  11.     "f0": data[:, 0],
  12. })
  14. data = dataframeToOperator(df_data, schemaStr='f0 string', op_type="batch")
  16. stringIndexer = StringIndexer() \
  17.     .setModelName("string_indexer_model") \
  18.     .setSelectedCol("f0") \
  19.     .setOutputCol("f0_indexed") \
  20.     .setStringOrderType("frequency_asc")
  22. indexed = stringIndexer.fit(data).transform(data)
  24. indexToString = IndexToString() \
  25.     .setModelName("string_indexer_model") \
  26.     .setSelectedCol("f0_indexed") \
  27.     .setOutputCol("f0_indxed_unindexed")
  29. indexToString.transform(indexed).print()

Results

  1. f0|f0_indexed|f0_indxed_unindexed
  2. --|----------|-------------------
  3. football|2|football
  4. football|2|football
  5. football|2|football
  6. basketball|1|basketball
  7. basketball|1|basketball
  8. tennis|0|tennis