| author | Christian Urban <urbanc@in.tum.de> |
| Thu, 06 Dec 2018 13:15:28 +0000 | |
| changeset 229 | cfcaf4a5e5b4 |
| parent 211 | 1859d978b18e |
| child 284 | fc20e5f83f0e |
| permissions | -rw-r--r-- |
| 211 | 1 |
|
2 |
import io.Source |
|
3 |
import scala.util._ |
|
4 |
||
5 |
def urban_get_csv_file(name: String) : List[String] = {
|
|
6 |
val csv = Source.fromFile(name) |
|
7 |
csv.mkString.split("\n").toList.drop(1)
|
|
8 |
} |
|
9 |
||
10 |
val urban_ratings = urban_get_csv_file("ratings.csv")
|
|
11 |
val urban_movies = urban_get_csv_file("movies.csv")
|
|
12 |
||
13 |
||
14 |
assert(process_ratings(urban_ratings).length == 48580) |
|
15 |
||
16 |
assert(process_movies(urban_movies).length == 9742) |
|
17 |
||
18 |