add file headers

Signed-off-by: Jason Wang <jasonwang_83@hotmail.com>
memoryz · Oct 26, 2023 · 03b8bbc · 03b8bbc
1 parent 1a2e320
commit 03b8bbc
Show file tree

Hide file tree

Showing 14 changed files with 61 additions and 10 deletions.
diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/BaseDiffInDiffEstimator.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/BaseDiffInDiffEstimator.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal
 
 import com.microsoft.azure.synapse.ml.causal.linalg.DVector
@@ -41,7 +44,10 @@ abstract class BaseDiffInDiffEstimator(override val uid: String)
 
   private[causal] val interactionCol = "interaction"
 
-  private[causal] def fitLinearModel(df: DataFrame, featureCols: Array[String], fitIntercept: Boolean, weightCol: Option[String] = None) = {
+  private[causal] def fitLinearModel(df: DataFrame,
+                                     featureCols: Array[String],
+                                     fitIntercept: Boolean,
+                                     weightCol: Option[String] = None) = {
     val assembler = new VectorAssembler()
       .setInputCols(featureCols)
       .setOutputCol("features")
@@ -151,4 +157,4 @@ class DiffInDiffModel(override val uid: String)
   }
 }
 
-object DiffInDiffModel extends ComplexParamsReadable[DiffInDiffModel]
+object DiffInDiffModel extends ComplexParamsReadable[DiffInDiffModel]
diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/CacheOps.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/CacheOps.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal
 
 import breeze.linalg.{DenseVector => BDV}

diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/DiffInDiffEstimator.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/DiffInDiffEstimator.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal
 
 import com.microsoft.azure.synapse.ml.codegen.Wrappable
@@ -30,7 +33,11 @@ class DiffInDiffEstimator(override val uid: String)
       )
       .withColumn(interactionCol, treatment * postTreatment)
 
-    val linearModel = fitLinearModel(didData, Array(getPostTreatmentCol, getTreatmentCol, interactionCol), fitIntercept = true)
+    val linearModel = fitLinearModel(
+      didData,
+      Array(getPostTreatmentCol, getTreatmentCol, interactionCol),
+      fitIntercept = true
+    )
 
     val treatmentEffect = linearModel.coefficients(2)
     val standardError = linearModel.summary.coefficientStandardErrors(2)
@@ -42,4 +49,4 @@ class DiffInDiffEstimator(override val uid: String)
   }, dataset.columns.length)
 }
 
-object DiffInDiffEstimator extends ComplexParamsReadable[DiffInDiffEstimator]
+object DiffInDiffEstimator extends ComplexParamsReadable[DiffInDiffEstimator]
diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/DiffInDiffEstimatorParams.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/DiffInDiffEstimatorParams.scala
@@ -1,8 +1,11 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal
 
 import org.apache.spark.ml.param.Params
 
 trait DiffInDiffEstimatorParams extends Params
   with HasTreatmentCol
   with HasOutcomeCol
-  with HasPostTreatmentCol
+  with HasPostTreatmentCol
diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/SharedParams.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/SharedParams.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal
 
 import org.apache.spark.ml.param.{Param, Params}

diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/SyntheticControlEstimator.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/SyntheticControlEstimator.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal
 
 import com.microsoft.azure.synapse.ml.codegen.Wrappable

diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/SyntheticDiffInDiffEstimator.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/SyntheticDiffInDiffEstimator.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal
 
 import com.microsoft.azure.synapse.ml.codegen.Wrappable
@@ -118,4 +121,4 @@ class SyntheticDiffInDiffEstimator(override val uid: String)
   }, dataset.columns.length)
 }
 
-object SyntheticDiffInDiffEstimator extends ComplexParamsReadable[SyntheticDiffInDiffEstimator]
+object SyntheticDiffInDiffEstimator extends ComplexParamsReadable[SyntheticDiffInDiffEstimator]
diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/SyntheticEstimator.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/SyntheticEstimator.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal
 
 import org.apache.spark.sql.expressions.{UserDefinedFunction, Window}
@@ -227,4 +230,4 @@ object SyntheticEstimator {
     // the regularization term will be incorrect.
     assignRowIndex(data.select(col(inputCol)).distinct.orderBy(col(inputCol)), indexCol)
   }
-}
+}
diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/SyntheticEstimatorParams.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/SyntheticEstimatorParams.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal
 
 import org.apache.spark.ml.param.{IntParam, LongParam, Param, ParamValidators, Params}

diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/linalg/MatrixOps.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/linalg/MatrixOps.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal.linalg
 
 import breeze.linalg.{DenseMatrix => BDM, DenseVector => BDV}
@@ -133,4 +136,4 @@ object BzMatrixOps extends MatrixOps[BDM[Double], BDV[Double]] {
   }
 
   override def size(matrix: BDM[Double]): (Long, Long) = (matrix.rows, matrix.cols)
-}
+}
diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/linalg/VectorOps.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/linalg/VectorOps.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal.linalg
 
 import breeze.linalg.{norm, DenseVector => BDV, max => bmax, sum => bsum}

diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/linalg/package.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/linalg/package.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal
 
 import breeze.linalg.{DenseMatrix => BDM, DenseVector => BDV, SparseVector => BSV}

diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/opt/ConstrainedLeastSquare.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/opt/ConstrainedLeastSquare.scala
@@ -1,9 +1,13 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal.opt
 
 import breeze.optimize.DiffFunction
 import com.microsoft.azure.synapse.ml.causal.CacheOps
 import com.microsoft.azure.synapse.ml.causal.linalg.{MatrixOps, VectorOps}
 
+// scalastyle:off non.ascii.character.disallowed
 /**
   * Solver for the following constrained least square problem:
   * minimize ||Ax-b||^2^ + λ||x||^2^, s.t. 1^T^x = 1, 0 ≤ x ≤ 1
@@ -12,6 +16,7 @@ import com.microsoft.azure.synapse.ml.causal.linalg.{MatrixOps, VectorOps}
   * @param numIterNoChange max number of iteration without change in loss function allowed before termination.
   * @param tol tolerance for loss function
   */
+// scalastyle:on
 private[causal] class ConstrainedLeastSquare[TMat, TVec](step: Double,
                                          maxIter: Int,
                                          numIterNoChange: Option[Int] = None,
@@ -75,4 +80,4 @@ private[causal] class ConstrainedLeastSquare[TMat, TVec](step: Double,
       (x, 0d, md.history.map(_.valueAt))
     }
   }
-}
+}
diff --git a/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/opt/MirrorDescent.scala b/core/src/main/scala/com/microsoft/azure/synapse/ml/causal/opt/MirrorDescent.scala
@@ -1,3 +1,6 @@
+// Copyright (C) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See LICENSE in project root for information.
+
 package com.microsoft.azure.synapse.ml.causal.opt
 
 import breeze.optimize.DiffFunction
@@ -133,4 +136,4 @@ private[opt] class MirrorDescent[TVec](private val func: DiffFunction[TVec],
     logger.debug("Elapsed time: " + (t1 - t0) / 1E9 + "s")
     result
   }
-}
+}