Skip to content
This repository has been archived by the owner on Nov 22, 2022. It is now read-only.

Commit

Permalink
Merge branch 'master' into SPARK-29464
Browse files Browse the repository at this point in the history
  • Loading branch information
zero323 authored Jan 12, 2020
2 parents 382744c + eee87b0 commit 9d79b64
Showing 1 changed file with 10 additions and 3 deletions.
13 changes: 10 additions & 3 deletions third_party/3/pyspark/ml/feature.pyi
Original file line number Diff line number Diff line change
Expand Up @@ -11,12 +11,19 @@ from pyspark.ml.wrapper import JavaEstimator, JavaModel, JavaParams, JavaTransf
from pyspark.ml.linalg import Vector, DenseVector, DenseMatrix
from pyspark.sql.dataframe import DataFrame

class Binarizer(JavaTransformer, HasInputCol, HasOutputCol, JavaMLReadable[Binarizer], JavaMLWritable):
class Binarizer(JavaTransformer, HasThreshold, HasThresholds, HasInputCol, HasOutputCol, HasInputCols, HasOutputCols, JavaMLReadable[Binarizer], JavaMLWritable):
threshold: Param[float]
def __init__(self, *, threshold: float = ..., inputCol: Optional[str] = ..., outputCol: Optional[str] = ...) -> None: ...
thresholds: Param[List[float]]
@overload
def __init__(self, *, inputCol: Optional[str] = ..., outputCol: Optional[str] = ...) -> None: ...
@overload
def __init__(self, *, threshold: float = ..., thresholds: Optional[List[float]] = ..., inputCols: Optional[List[str]] = ..., outputCols: Optional[List[str]] = ...) -> None: ...
@overload
def setParams(self, *, threshold: float = ..., inputCol: Optional[str] = ..., outputCol: Optional[str] = ...) -> Binarizer: ...
@overload
def setParams(self, *, thresholds: Optional[List[float]] = ..., inputCols: Optional[List[str]] = ..., outputCols: Optional[List[str]] = ...) -> Binarizer: ...
def setThreshold(self, value: float) -> Binarizer: ...
def getThreshold(self) -> float: ...
def setThresholds(self, value: List[float]) -> Binarizer: ...

class _LSHParams(HasInputCol, HasOutputCol):
numHashTables: Param[int]
Expand Down

0 comments on commit 9d79b64

Please sign in to comment.