| author | Christian Urban <christian.urban@kcl.ac.uk> |
| Sun, 31 Jan 2021 03:28:20 +0000 | |
| changeset 393 | a8e7e187bb26 |
| parent 379 | 73ad2e018516 |
| child 403 | 312c9eb39ad8 |
| permissions | -rw-r--r-- |
| 211 | 1 |
|
2 |
import io.Source |
|
3 |
import scala.util._ |
|
4 |
||
| 284 | 5 |
import CW7b._ |
6 |
||
| 211 | 7 |
def urban_get_csv_file(name: String) : List[String] = {
|
| 379 | 8 |
val csv = Source.fromFile(name)("ISO-8859-1")
|
| 211 | 9 |
csv.mkString.split("\n").toList.drop(1)
|
10 |
} |
|
11 |
||
12 |
val urban_ratings = urban_get_csv_file("ratings.csv")
|
|
13 |
val urban_movies = urban_get_csv_file("movies.csv")
|
|
14 |
||
15 |
||
16 |
assert(process_ratings(urban_ratings).length == 48580) |
|
17 |
||
18 |
assert(process_movies(urban_movies).length == 9742) |
|
19 |
||
20 |