| author | Christian Urban <christian.urban@kcl.ac.uk> |
| Mon, 30 Nov 2020 00:06:15 +0000 | |
| changeset 379 | 5616b45d656f |
| parent 347 | 4de31fdc0d67 |
| child 403 | ffce7b61b446 |
| 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 |