remove duplicate in stream close

This commit is contained in:
CodingCat 2016-03-09 12:33:49 -05:00
parent 852c5a4b32
commit 005b1276d0

View File

@ -107,7 +107,6 @@ object XGBoost extends Serializable {
def loadModelFromHadoop(modelPath: String)(implicit sparkContext: SparkContext): XGBoostModel = { def loadModelFromHadoop(modelPath: String)(implicit sparkContext: SparkContext): XGBoostModel = {
val dataInStream = FileSystem.get(sparkContext.hadoopConfiguration).open(new Path(modelPath)) val dataInStream = FileSystem.get(sparkContext.hadoopConfiguration).open(new Path(modelPath))
val xgBoostModel = new XGBoostModel(SXGBoost.loadModel(dataInStream)) val xgBoostModel = new XGBoostModel(SXGBoost.loadModel(dataInStream))
dataInStream.close()
xgBoostModel xgBoostModel
} }
} }