>백엔드 개발 >파이썬 튜토리얼 >기계 학습에 PySpark를 사용하는 방법

기계 학습에 PySpark를 사용하는 방법

DDD
DDD원래의
2024-12-11 08:47:10834검색

Apache Spark(빅 데이터 처리를 위한 오픈 소스 프레임워크) 출시 이후 여러 컨테이너에 걸쳐 대용량 데이터를 병렬로 처리하는 데 가장 널리 사용되는 기술 중 하나가 되었습니다. 효율성과 속도에 자부심을 갖고 있습니다. 이전에 존재했던 유사한 소프트웨어와 비교해 보세요.

Python 프로그래밍 언어를 사용하여 ApacheSpark의 놀라운 잠재력과 상호 작용하고 활용할 수 있게 해주는 Python API인 PySpark를 통해 Python에서 이 놀라운 기술을 사용하는 것이 가능합니다.

이 기사에서는 PySpark를 사용하여 선형 회귀 알고리즘을 사용하여 기계 학습 모델을 구축하는 방법을 배우고 시작합니다.

참고: 이 문서에 포함된 개념을 올바르게 이해하려면 Python, VSCode와 같은 IDE, 명령 프롬프트/터미널 사용 방법, 기계 학습 개념에 대한 사전 지식이 필수적입니다.

이 기사를 통해 다음을 수행할 수 있습니다.

  • ApacheSpark가 무엇인지 이해하세요.
  • PySpark에 대해 알아보고 이를 기계 학습에 사용하는 방법을 알아보세요.

PySpark란 무엇인가요?

Apache Spark 공식 웹사이트에 따르면 PySpark를 사용하면 "데이터 엔지니어링을 위해 ApacheSpark(단순성, 속도, 확장성, 다양성)와 Python(풍부한 생태계, 성숙한 라이브러리, 단순성)의 결합된 장점을 활용할 수 있습니다. , 데이터 과학, 단일 노드 머신 또는 클러스터에서의 머신 러닝.”

How to Use PySpark for Machine Learning
이미지 출처

PySpark는 ApacheSpark용 Python API입니다. 즉, Python으로 작성된 코드가 Scala로 작성된 ApacheSpark 기술과 통신할 수 있게 해주는 인터페이스 역할을 합니다. 이러한 방식으로 이미 Python 생태계에 익숙한 전문가는 ApacheSpark 기술을 빠르게 활용할 수 있습니다. 이는 또한 Python에서 사용되는 기존 라이브러리가 관련성을 유지하도록 보장합니다.

기계 학습을 위해 PySpark를 사용하는 방법에 대한 자세한 가이드

다음 단계에서는 선형 회귀 알고리즘을 사용하여 기계 학습 모델을 구축합니다.

  • 프로젝트 종속성 설치: 이미 컴퓨터에 Python이 설치되어 있다고 가정합니다. 그렇지 않은 경우 다음 단계로 이동하기 전에 설치하세요. 터미널이나 명령 프롬프트를 열고 아래 코드를 입력하여 PySpark 라이브러리를 설치하세요.
pip install pyspark

이러한 추가 Python 라이브러리가 없으면 설치할 수 있습니다.

pip install pyspark
  • 파일 생성 및 필요한 라이브러리 가져오기: VSCode를 열고 선택한 프로젝트 디렉터리에서 프로젝트용 파일(예: pyspart_model.py)을 생성합니다. 파일을 열고 프로젝트에 필요한 라이브러리를 가져옵니다.
pip install pandas numpy
  • Spark 세션 만들기: imports 아래에 이 코드를 입력하여 프로젝트에 대한 Spark 세션을 시작합니다.
from pyspark.sql import SparkSession
from pyspark.ml.feature import VectorAssembler
from pyspark.ml.classification import LogisticRegression
from pyspark.ml.evaluation import BinaryClassificationEvaluator
import pandas as pd
  • CSV 파일(작업할 데이터세트) 읽기: 프로젝트 디렉터리/폴더에 data.csv라는 데이터세트가 이미 있는 경우 아래 코드를 사용하여 로드하세요.
spark = SparkSession.builder.appName("LogisticRegressionExample").getOrCreate()
  • 탐색적 데이터 분석: 이 단계는 작업 중인 데이터 세트를 이해하는 데 도움이 됩니다. null 값을 확인하고 사용할 정리 접근 방식을 결정합니다.
data = spark.read.csv("data.csv", header=True, inferSchema=True)

선택적으로 소규모 데이터 세트로 작업하는 경우 이를 Python 데이터 프레임 및 디렉터리로 변환하고 Python을 사용하여 누락된 값을 확인할 수 있습니다.

# Display the schema my
 data.printSchema() 
# Show the first ten rows 
data.show(10)
# Count null values in each column
missing_values = df.select(
    [count(when(isnull(c), c)).alias(c) for c in df.columns]
)

# Show the result
missing_values.show()
  • 데이터 전처리: 이 단계에는 데이터세트의 열/특성을 PySpark의 기계 학습 라이브러리가 쉽게 이해할 수 있거나 호환되는 형식으로 변환하는 작업이 포함됩니다.

VectorAssembler를 사용하여 모든 기능을 단일 벡터 열로 결합합니다.

pandas_df = data.toPandas()
# Use Pandas to check missing values
print(pandas_df.isna().sum())
  • 데이터세트 분할: 데이터세트를 편리한 비율로 분할합니다. 여기서는 70~30%를 사용합니다. 70%는 학습에, 30%는 모델 테스트에 사용됩니다.
# Combine feature columns into a single vector column
feature_columns = [col for col in data.columns if col != "label"]
assembler = VectorAssembler(inputCols=feature_columns, outputCol="features")

# Transform the data
data = assembler.transform(data)

# Select only the 'features' and 'label' columns for training
final_data = data.select("features", "label")

# Show the transformed data
final_data.show(5)
  • 모델 교육: 모델 교육에 로지스틱 회귀 알고리즘을 사용하고 있습니다.

LogisticRegression 클래스의 인스턴스를 생성하고 모델에 적합합니다.

train_data, test_data = final_data.randomSplit([0.7, 0.3], seed=42)
  • 훈련된 모델로 예측: 이전 단계에서 훈련한 모델을 사용하여 예측합니다.
lr = LogisticRegression(featuresCol="features", labelCol="label")

# Train the model
lr_model = lr.fit(train_data)
  • 모델 평가: 여기에서는 예측 성능이나 정확성 수준을 결정하기 위해 모델을 평가합니다. 우리는 적절한 평가 지표를 사용하여 이를 달성합니다.

AUC 측정항목을 사용하여 모델 평가

predictions = lr_model.transform(test_data)
# Show predictions
predictions.select("features", "label", "prediction", "probability").show(5)

이 기사에 사용된 엔드투엔드 코드는 다음과 같습니다.

evaluator = BinaryClassificationEvaluator(rawPredictionCol="rawPrediction", labelCol="label", metricName="areaUnderROC")

# Compute the AUC
auc = evaluator.evaluate(predictions)
print(f"Area Under ROC: {auc}")

다음 단계는?

이 글이 끝났습니다. 위 단계에 따라 PySpark를 사용하여 머신러닝 모델을 구축했습니다.

다음 단계를 진행하기 전에 항상 데이터세트가 깨끗하고 null 값이 없는지 확인하세요. 마지막으로, 모델 학습을 진행하기 전에 모든 기능에 숫자 값이 포함되어 있는지 확인하세요.

위 내용은 기계 학습에 PySpark를 사용하는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!

성명:
본 글의 내용은 네티즌들의 자발적인 기여로 작성되었으며, 저작권은 원저작자에게 있습니다. 본 사이트는 이에 상응하는 법적 책임을 지지 않습니다. 표절이나 침해가 의심되는 콘텐츠를 발견한 경우 admin@php.cn으로 문의하세요.