gpt4 book ai didi

apache-spark - 星火 ML : Taking square root of feature columns

转载 作者:行者123 更新时间:2023-12-04 05:20:30 24 4
gpt4 key购买 nike

您好,我正在使用自定义 UDF 对每列中的每个值求平方根。

square_root_UDF = udf(lambda x: math.sqrt(x), DoubleType())

for x in features:
dataTraining = dataTraining.withColumn(x, square_root_UDF(x))

有没有更快的方法来完成它?多项式展开函数不适用于这种情况。

最佳答案

不要使用 UDF。而是使用内置的:

from pyspark.sql.functions import sqrt

for x in features:
dataTraining = dataTraining.withColumn(x, sqrt(x))

关于apache-spark - 星火 ML : Taking square root of feature columns,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42979735/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com