changed filter relation on SubRelType

This commit is contained in:
Sandro La Bruzzo 2022-07-21 10:11:48 +02:00
parent 5b76321d9c
commit 5f651f2316
1 changed files with 1 additions and 3 deletions

View File

@ -112,13 +112,11 @@ object SparkConvertRDDtoDataset {
"opencitations".equalsIgnoreCase(k.getValue) "opencitations".equalsIgnoreCase(k.getValue)
) )
) )
.filter(r => r.getRelClass != null && r.getRelClass.equalsIgnoreCase(filterRelation)) .filter(r => r.getSubRelType != null && r.getSubRelType.equalsIgnoreCase(filterRelation))
spark.createDataset(rddRelation).as[Relation].write.mode(SaveMode.Overwrite).save(s"$relPath") spark.createDataset(rddRelation).as[Relation].write.mode(SaveMode.Overwrite).save(s"$relPath")
} else { } else {
val relationSemanticFilter = List( val relationSemanticFilter = List(
// "cites",
// "iscitedby",
"merges", "merges",
"ismergedin", "ismergedin",
"HasAmongTopNSimilarDocuments", "HasAmongTopNSimilarDocuments",