Классификация текста

Введение в Spark SQL на Python

Mark Plutowski

Data Scientist

Предсказание конечного слова

Введение в Spark SQL на Python

Стрелка последовательности

Введение в Spark SQL на Python

Конечное слово

Введение в Spark SQL на Python

Конечное слово со скобкой

Введение в Spark SQL на Python

Перемешивание 1

Введение в Spark SQL на Python

Перемешивание 2

Введение в Spark SQL на Python

Песни

Введение в Spark SQL на Python

Видео

Введение в Spark SQL на Python

Выборка данных

df_true = df.where("endword in ('she', 'he', 'hers', 'his', 'her', 'him')")\
            .withColumn('label', lit(1))


df_false = df.where("endword not in ('she', 'he', 'hers', 'his', 'her', 'him')")\
           .withColumn('label', lit(0))
Введение в Spark SQL на Python

Объединение положительных и отрицательных примеров

df_examples = df_true.union(df_false)
Введение в Spark SQL на Python

Разделение данных на обучающую и оценочную выборки

df_train, df_eval = df_examples.randomSplit((0.60, 0.40), 42)
Введение в Spark SQL на Python

Обучение

from pyspark.ml.classification import LogisticRegression

logistic = LogisticRegression(maxIter=50, regParam=0.6, elasticNetParam=0.3)

model = logistic.fit(df_train)

print("Training iterations: ", model.summary.totalIterations)
Введение в Spark SQL на Python

Давайте потренируемся!

Введение в Spark SQL на Python

Preparing Video For Download...