main_testing2/danube_test2.scala
author Christian Urban <christian.urban@kcl.ac.uk>
Mon, 08 Nov 2021 02:16:33 +0000
changeset 406 fcf31ffb2dd7
parent 403 312c9eb39ad8
child 419 7268c66cf77c
permissions -rw-r--r--
updated
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
211
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     1
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     2
import io.Source
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     3
import scala.util._
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     4
403
312c9eb39ad8 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 379
diff changeset
     5
import M2._
284
fc20e5f83f0e updated
Christian Urban <urbanc@in.tum.de>
parents: 211
diff changeset
     6
211
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     7
def urban_get_csv_file(name: String) : List[String] = {
379
73ad2e018516 updated
Christian Urban <christian.urban@kcl.ac.uk>
parents: 347
diff changeset
     8
  val csv = Source.fromFile(name)("ISO-8859-1")
211
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
     9
  csv.mkString.split("\n").toList.drop(1)
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    10
}
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    11
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    12
val urban_ratings = urban_get_csv_file("ratings.csv")
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    13
val urban_movies = urban_get_csv_file("movies.csv")
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    14
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    15
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    16
assert(process_ratings(urban_ratings).length == 48580)
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    17
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    18
assert(process_movies(urban_movies).length == 9742)
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    19
1859d978b18e updated
Christian Urban <urbanc@in.tum.de>
parents:
diff changeset
    20