87 lines
2.8 KiB
Groovy
87 lines
2.8 KiB
Groovy
apply plugin: 'java'
|
|
|
|
ext.lenskitVersion = '3.0-M1'
|
|
if (!hasProperty('dataDir')) {
|
|
ext.dataDir = 'data'
|
|
}
|
|
|
|
sourceCompatibility = 1.7
|
|
|
|
apply from: "$rootDir/gradle/repositories.gradle"
|
|
|
|
dependencies {
|
|
compile "org.lenskit:lenskit-core:$lenskitVersion"
|
|
runtime "org.lenskit:lenskit-cli:$lenskitVersion"
|
|
}
|
|
dependencies {
|
|
testCompile group: 'junit', name: 'junit', version: '4.11'
|
|
}
|
|
|
|
task runMean(type: JavaExec, group: 'run') {
|
|
description "Run the simple mean recommender."
|
|
classpath sourceSets.main.runtimeClasspath
|
|
main 'org.lenskit.cli.Main'
|
|
args '--log-file', file("$buildDir/recommend-mean.log"), '--log-file-level', 'DEBUG'
|
|
args 'global-recommend'
|
|
args '--data-source', "$dataDir/movielens.yml"
|
|
args '-c', file('etc/mean.groovy')
|
|
args '-n', 10
|
|
if (project.hasProperty('lenskit.maxMemory')) {
|
|
maxHeapSize project.getProperty('lenskit.maxMemory')
|
|
}
|
|
}
|
|
|
|
task runDampedMean(type: JavaExec, group: 'run') {
|
|
description "Run the damped mean recommender."
|
|
mustRunAfter runMean
|
|
classpath sourceSets.main.runtimeClasspath
|
|
main 'org.lenskit.cli.Main'
|
|
args '--log-file', file("$buildDir/recommend-damped-mean.log"), '--log-file-level', 'DEBUG'
|
|
args 'global-recommend'
|
|
args '--data-source', "$dataDir/movielens.yml"
|
|
args '-c', file('etc/damped-mean.groovy')
|
|
if (project.hasProperty('lenskit.maxMemory')) {
|
|
maxHeapSize project.getProperty('lenskit.maxMemory')
|
|
}
|
|
}
|
|
|
|
task runBasicAssoc(type: JavaExec, group: 'run') {
|
|
description "Run the damped mean recommender."
|
|
mustRunAfter runDampedMean
|
|
classpath sourceSets.main.runtimeClasspath
|
|
main 'org.lenskit.cli.Main'
|
|
args '--log-file', file("$buildDir/recommend-basic-assoc.log"), '--log-file-level', 'DEBUG'
|
|
args 'global-recommend'
|
|
args '--data-source', "$dataDir/movielens.yml"
|
|
args '-c', file('etc/simple-assoc.groovy')
|
|
args findProperty('referenceItem') ?: 260
|
|
if (project.hasProperty('lenskit.maxMemory')) {
|
|
maxHeapSize project.getProperty('lenskit.maxMemory')
|
|
}
|
|
}
|
|
|
|
task runLiftAssoc(type: JavaExec, group: 'run') {
|
|
description "Run the damped mean recommender."
|
|
classpath sourceSets.main.runtimeClasspath
|
|
mustRunAfter runBasicAssoc
|
|
main 'org.lenskit.cli.Main'
|
|
args '--log-file', file("$buildDir/recommend-lift-assoc.log"), '--log-file-level', 'DEBUG'
|
|
args 'global-recommend'
|
|
args '--data-source', "$dataDir/movielens.yml"
|
|
args '-c', file('etc/lift-assoc.groovy')
|
|
args findProperty('referenceItem') ?: 2761
|
|
if (project.hasProperty('lenskit.maxMemory')) {
|
|
maxHeapSize project.getProperty('lenskit.maxMemory')
|
|
}
|
|
}
|
|
|
|
task runAll(group: 'run') {
|
|
dependsOn runMean, runDampedMean
|
|
dependsOn runBasicAssoc, runLiftAssoc
|
|
}
|
|
|
|
task prepareSubmission(type: Copy) {
|
|
from jar
|
|
into distsDir
|
|
rename(/-assignment/, '-submission')
|
|
}
|