20 parser.add_argument('--inputDir',\ |
20 parser.add_argument('--inputDir',\ |
21 help='Directory containing all the input data. MaSh expects the following files: mash_features,mash_dependencies,mash_accessibility') |
21 help='Directory containing all the input data. MaSh expects the following files: mash_features,mash_dependencies,mash_accessibility') |
22 parser.add_argument('--depFile', default='mash_dependencies', |
22 parser.add_argument('--depFile', default='mash_dependencies', |
23 help='Name of the file with the premise dependencies. The file must be in inputDir. Default = mash_dependencies') |
23 help='Name of the file with the premise dependencies. The file must be in inputDir. Default = mash_dependencies') |
24 |
24 |
25 parser.add_argument('--algorithm',default='nb',help="Which learning algorithm is used. nb = Naive Bayes,predef=predefined. Default=nb.") |
25 parser.add_argument('--algorithm',default='nb',help="Which learning algorithm is used. nb = Naive Bayes,KNN,predef=predefined. Default=nb.") |
|
26 parser.add_argument('--predef',help="File containing the predefined suggestions. Only used when algorithm = predef.") |
26 # NB Parameters |
27 # NB Parameters |
27 parser.add_argument('--NBDefaultPriorWeight',default=20.0,help="Initializes classifiers with value * p |- p. Default=20.0.",type=float) |
28 parser.add_argument('--NBDefaultPriorWeight',default=20.0,help="Initializes classifiers with value * p |- p. Default=20.0.",type=float) |
28 parser.add_argument('--NBDefVal',default=-15.0,help="Default value for unknown features. Default=-15.0.",type=float) |
29 parser.add_argument('--NBDefVal',default=-15.0,help="Default value for unknown features. Default=-15.0.",type=float) |
29 parser.add_argument('--NBPosWeight',default=10.0,help="Weight value for positive features. Default=10.0.",type=float) |
30 parser.add_argument('--NBPosWeight',default=10.0,help="Weight value for positive features. Default=10.0.",type=float) |
30 # TODO: Rename to sineFeatures |
31 parser.add_argument('--expandFeatures',default=False,action='store_true',help="Learning-based feature expansion. Default=False.") |
31 parser.add_argument('--sineFeatures',default=False,action='store_true',help="Uses a SInE like prior for premise lvl predictions. Default=False.") |
|
32 parser.add_argument('--sineWeight',default=0.5,help="How much the SInE prior is weighted. Default=0.5.",type=float) |
|
33 |
32 |
34 parser.add_argument('--statistics',default=False,action='store_true',help="Create and show statistics for the top CUTOFF predictions.\ |
33 parser.add_argument('--statistics',default=False,action='store_true',help="Create and show statistics for the top CUTOFF predictions.\ |
35 WARNING: This will make the program a lot slower! Default=False.") |
34 WARNING: This will make the program a lot slower! Default=False.") |
36 parser.add_argument('--saveStats',default=None,help="If defined, stores the statistics in the filename provided.") |
35 parser.add_argument('--saveStats',default=None,help="If defined, stores the statistics in the filename provided.") |
37 parser.add_argument('--cutOff',default=500,help="Option for statistics. Only consider the first cutOff predictions. Default=500.",type=int) |
36 parser.add_argument('--cutOff',default=500,help="Option for statistics. Only consider the first cutOff predictions. Default=500.",type=int) |