added main method
This commit is contained in:
parent
68ba9e83cb
commit
6a8386e897
|
@ -105,7 +105,8 @@ class DecisionTree:
|
|||
# sum(f1_scores_train)/float(len(f1_scores_train))))
|
||||
# print()
|
||||
|
||||
#################################
|
||||
if __name__ == '__main__':
|
||||
|
||||
print('# starting decision tree')
|
||||
print('# ...')
|
||||
|
||||
|
|
|
@ -159,7 +159,8 @@ class NaiveBayes:
|
|||
#print metrics
|
||||
print('F1 score: ', format(f1_score(y_train_test, predictions)))
|
||||
|
||||
#################################
|
||||
if __name__ == '__main__':
|
||||
|
||||
print('# starting naive bayes')
|
||||
print('# ...')
|
||||
|
||||
|
|
|
@ -16,6 +16,10 @@ graphviz==0.9
|
|||
|
||||
scikit_learn==0.19.2
|
||||
|
||||
## Installation under (UBUNTU?)
|
||||
## Installation under Windows
|
||||
|
||||
pip install XY
|
||||
|
||||
## Installation under UBUNTU
|
||||
|
||||
apt-get install XX
|
||||
|
|
|
@ -94,6 +94,8 @@ class Requester:
|
|||
# save csv
|
||||
CsvHandler.write_csv(df, filestring)
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
print('# starting requester')
|
||||
print('# ...')
|
||||
save_articles_from_webhoseio()
|
||||
|
|
Loading…
Reference in New Issue