Skip to content
This repository has been archived by the owner on Mar 1, 2018. It is now read-only.

Commit

Permalink
Merge branch 'master' into cuducos_enhance_irregular_companies_tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@anaschwendler
anaschwendler committed Dec 23, 2017
2 parents 83dde70 + bcbff22 commit d7ec25c
Show file tree
Hide file tree
Showing 2 changed files with 109 additions and 24 deletions.
73 changes: 61 additions & 12 deletions rosie/chamber_of_deputies/classifiers/monthly_subquota_limit_classifier.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -42,33 +42,82 @@ def fit(self, X):
def transform(self, X=None):
self.limits = [
{
# Automotive vehicle renting or charter (From 04/2015 to 06/2017)
'data': self._X.query('(subquota_number == "120") & (reimbursement_month >= datetime(2015, 4, 1)) & (reimbursement_month <= datetime(2017, 6, 1))'),
# Automotive vehicle renting or charter (From 12/2013 to 03/2015)
'data': self._X.query('(subquota_number == "120") & '
'(reimbursement_month >= datetime(2013, 12, 1)) & '
'(reimbursement_month <= datetime(2015, 3, 1))'),
'monthly_limit': 1000000,
},
{
# Automotive vehicle renting or charter (From 04/2015 to 04/2017)
'data': self._X.query('(subquota_number == "120") & '
'(reimbursement_month >= datetime(2015, 4, 1)) & '
'(reimbursement_month <= datetime(2017, 4, 1))'),
'monthly_limit': 1090000,
},
{
# Automotive vehicle renting or charter (From 07/2017)
'data': self._X.query('(subquota_number == "120") & (reimbursement_month >= datetime(2017, 7, 1))'),
# Automotive vehicle renting or charter (From 05/2017)
'data': self._X.query('(subquota_number == "120") & '
'(reimbursement_month >= datetime(2017, 5, 1))'),
'monthly_limit': 1271300,
},
{
# Taxi, toll and parking
'data': self._X.query('(subquota_number == "122") & (reimbursement_month >= datetime(2015, 4, 1))'),
# Taxi, toll and parking (From 12/2013 to 03/2015)
'data': self._X.query('(subquota_number == "122") & '
'(reimbursement_month >= datetime(2013, 12, 1)) & '
'(reimbursement_month <= datetime(2015, 3, 1))'),
'monthly_limit': 250000,
},
{
# Taxi, toll and parking (From 04/2015)
'data': self._X.query('(subquota_number == "122") & '
'(reimbursement_month >= datetime(2015, 4, 1))'),
'monthly_limit': 270000,
},
{
# Fuels and lubricants
'data': self._X.query('(subquota_number == "3") & (reimbursement_month >= datetime(2015, 10, 1))'),
# Fuels and lubricants (From 07/2009 to 03/2015)
'data': self._X.query('(subquota_number == "3") & '
'(reimbursement_month >= datetime(2009, 7, 1)) & '
'(reimbursement_month <= datetime(2015, 3, 1))'),
'monthly_limit': 450000,
},
{
# Fuels and lubricants (From 04/2015 to 08/2015)
'data': self._X.query('(subquota_number == "3") & '
'(reimbursement_month >= datetime(2015, 4, 1)) & '
'(reimbursement_month <= datetime(2015, 8, 1))'),
'monthly_limit': 490000,
},
{
# Fuels and lubricants (From 9/2015)
'data': self._X.query('(subquota_number == "3") & '
'(reimbursement_month >= datetime(2015, 9, 1))'),
'monthly_limit': 600000,
},
{
# Security service provided by specialized company
'data': self._X.query('(subquota_number == "8") & (reimbursement_month >= datetime(2015, 4, 1))'),
# Security service provided by specialized company (From 07/2009 to 4/2014)
'data': self._X.query('(subquota_number == "8") & '
'(reimbursement_month >= datetime(2009, 7, 1)) & '
'(reimbursement_month <= datetime(2014, 4, 1))'),
'monthly_limit': 450000,
},
{
# Security service provided by specialized company (From 05/2014 to 3/2015)
'data': self._X.query('(subquota_number == "8") & '
'(reimbursement_month >= datetime(2014, 5, 1)) & '
'(reimbursement_month <= datetime(2015, 3, 1))'),
'monthly_limit': 800000,
},
{
# Security service provided by specialized company (From 04/2015)
'data': self._X.query('(subquota_number == "8") & '
'(reimbursement_month >= datetime(2015, 4, 1))'),
'monthly_limit': 870000,
},
{
# Participation in course, talk or similar event
'data': self._X.query('(subquota_number == "137") & (reimbursement_month >= datetime(2015, 11, 1))'),
# Participation in course, talk or similar event (From 10/2015)
'data': self._X.query('(subquota_number == "137") & '
'(reimbursement_month >= datetime(2015, 10, 1))'),
'monthly_limit': 769716,
},
]
Expand Down
60 changes: 48 additions & 12 deletions rosie/chamber_of_deputies/tests/fixtures/monthly_subquota_limit_classifier.csv
View file
Original file line number Diff line number Diff line change
@@ -1,42 +1,42 @@
applicant_id,subquota_number,issue_date,year,month,net_value,expected_prediction,test_case_description
1,120,2015-03-01,2015,3,10500,False,test_predict_false_when_not_in_date_range
1,120,2015-03-01,2015,3,401,False,test_predict_false_when_not_in_date_range
1,120,2013-11-01,2013,11,10500,False,test_predict_false_when_not_in_date_range
1,120,2013-11-01,2013,11,401,False,test_predict_false_when_not_in_date_range
2,120,2015-04-01,2015,4,10500,False,test_predict_false_when_under_the_limit
2,120,2015-04-01,2015,4,399,False,test_predict_false_when_under_the_limit
3,120,2015-05-01,2015,5,10500,False,test_predict_false_when_exactly_on_the_limit
3,120,2015-05-01,2015,5,400,False,test_predict_false_when_exactly_on_the_limit
4,120,2015-06-01,2015,6,10500,False,test_predict_true_when_over_the_limit
4,120,2015-06-01,2015,6,401,True,test_predict_true_when_over_the_limit
4,120,2015-06-01,2015,6,10,True,test_predict_true_when_over_the_limit
1,122,2015-03-01,2015,3,2000,False,test_predict_false_when_not_in_date_range
1,122,2015-03-01,2015,3,701,False,test_predict_false_when_not_in_date_range
1,122,2013-11-01,2013,11,2000,False,test_predict_false_when_not_in_date_range
1,122,2013-11-01,2013,11,701,False,test_predict_false_when_not_in_date_range
2,122,2015-04-01,2015,4,2000,False,test_predict_false_when_under_the_limit
2,122,2015-04-01,2015,4,699,False,test_predict_false_when_under_the_limit
3,122,2015-05-01,2015,5,2000,False,test_predict_false_when_exactly_on_the_limit
3,122,2015-05-01,2015,5,700,False,test_predict_false_when_exactly_on_the_limit
4,122,2015-06-01,2015,6,2000,False,test_predict_true_when_over_the_limit
4,122,2015-06-01,2015,6,701,True,test_predict_true_when_over_the_limit
4,122,2015-06-01,2015,6,10,True,test_predict_true_when_over_the_limit
1,3,2015-09-01,2015,9,5000,False,test_predict_false_when_not_in_date_range
1,3,2015-09-01,2015,9,1001,False,test_predict_false_when_not_in_date_range
1,3,2009-06-01,2009,6,5000,False,test_predict_false_when_not_in_date_range
1,3,2009-06-01,2009,6,1001,False,test_predict_false_when_not_in_date_range
2,3,2015-10-01,2015,10,5000,False,test_predict_false_when_under_the_limit
2,3,2015-10-01,2015,10,999,False,test_predict_false_when_under_the_limit
3,3,2015-11-01,2015,11,5000,False,test_predict_false_when_exactly_on_the_limit
3,3,2015-11-01,2015,11,1000,False,test_predict_false_when_exactly_on_the_limit
4,3,2015-12-01,2015,12,5000,False,test_predict_true_when_over_the_limit
4,3,2015-12-01,2015,12,1001,True,test_predict_true_when_over_the_limit
4,3,2015-12-01,2015,12,10,True,test_predict_true_when_over_the_limit
1,8,2015-03-01,2015,3,8000,False,test_predict_false_when_not_in_date_range
1,8,2015-03-01,2015,3,701,False,test_predict_false_when_not_in_date_range
1,8,2009-06-01,2009,6,8000,False,test_predict_false_when_not_in_date_range
1,8,2009-06-01,2009,6,701,False,test_predict_false_when_not_in_date_range
2,8,2015-04-01,2015,4,8000,False,test_predict_false_when_under_the_limit
2,8,2015-04-01,2015,4,699,False,test_predict_false_when_under_the_limit
3,8,2015-05-01,2015,5,8000,False,test_predict_false_when_exactly_on_the_limit
3,8,2015-05-01,2015,5,700,False,test_predict_false_when_exactly_on_the_limit
4,8,2015-06-01,2015,6,8000,False,test_predict_true_when_over_the_limit
4,8,2015-06-01,2015,6,701,True,test_predict_true_when_over_the_limit
4,8,2015-06-01,2015,6,10,True,test_predict_true_when_over_the_limit
1,137,2015-10-01,2015,10,7000,False,test_predict_false_when_not_in_date_range
1,137,2015-10-01,2015,10,698,False,test_predict_false_when_not_in_date_range
1,137,2015-09-01,2015,09,7000,False,test_predict_false_when_not_in_date_range
1,137,2015-09-01,2015,09,698,False,test_predict_false_when_not_in_date_range
2,137,2015-11-01,2015,11,7000,False,test_predict_false_when_under_the_limit
2,137,2015-11-01,2015,11,697,False,test_predict_false_when_under_the_limit
3,137,2015-12-01,2015,12,7000,False,test_predict_false_when_exactly_on_the_limit
Expand All @@ -46,5 +46,41 @@ applicant_id,subquota_number,issue_date,year,month,net_value,expected_prediction
4,137,2016-01-01,2016,1,10,True,test_predict_true_when_over_the_limit
100,120,2017-04-01,2017,4,12713,True,test_predict_true_when_over_the_limit
100,120,2017-04-01,2017,4,12713.01,True,test_predict_true_when_over_the_limit
100,120,2017-05-01,2017,5,12713,True,test_predict_true_when_exactly_on_the_limit
100,120,2017-05-01,2017,5,12713.01,True,test_predict_true_when_over_the_limit
100,120,2017-05-01,2017,5,12713,False,test_predict_false_when_exactly_on_the_limit
100,120,2017-05-01,2017,5,12713.01,True,test_predict_true_when_over_the_limit
5,120,2013-12-01,2013,12,9600,False,test_predict_false_when_under_the_limit
5,120,2013-12-01,2013,12,399,False,test_predict_false_when_under_the_limit
6,120,2014-02-01,2014,02,9000,False,test_predict_false_when_exactly_on_the_limit
6,120,2014-02-01,2014,02,1000,False,test_predict_false_when_exactly_on_the_limit
7,120,2014-03-01,2014,03,9999,False,test_predict_true_when_over_the_limit
7,120,2014-03-01,2014,03,100,True,test_predict_true_when_over_the_limit
5,122,2013-12-01,2013,12,2400,False,test_predict_false_when_under_the_limit
5,122,2013-12-01,2013,12,99,False,test_predict_false_when_under_the_limit
6,122,2014-02-01,2014,02,2400,False,test_predict_false_when_exactly_on_the_limit
6,122,2014-02-01,2014,02,100,False,test_predict_false_when_exactly_on_the_limit
7,122,2014-03-01,2014,03,2500,False,test_predict_true_when_over_the_limit
7,122,2014-03-01,2014,03,10,True,test_predict_true_when_over_the_limit
5,3,2013-12-01,2013,12,4300,False,test_predict_false_when_under_the_limit
5,3,2013-12-01,2013,12,199,False,test_predict_false_when_under_the_limit
6,3,2014-02-01,2014,02,4320,False,test_predict_false_when_exactly_on_the_limit
6,3,2014-02-01,2014,02,180,False,test_predict_false_when_exactly_on_the_limit
7,3,2014-03-01,2014,03,4499,False,test_predict_true_when_over_the_limit
7,3,2014-03-01,2014,03,2,True,test_predict_true_when_over_the_limit
5,3,2015-04-01,2015,04,4700,False,test_predict_false_when_under_the_limit
5,3,2015-04-01,2015,04,199,False,test_predict_false_when_under_the_limit
6,3,2015-05-01,2015,05,4720,False,test_predict_false_when_exactly_on_the_limit
6,3,2015-05-01,2015,05,180,False,test_predict_false_when_exactly_on_the_limit
7,3,2015-06-01,2015,06,4899,False,test_predict_true_when_over_the_limit
7,3,2015-06-01,2015,06,2,True,test_predict_true_when_over_the_limit
5,8,2009-07-01,2009,07,4498,False,test_predict_false_when_under_the_limit
5,8,2009-07-01,2009,07,1,False,test_predict_false_when_under_the_limit
6,8,2009-08-01,2009,08,4499,False,test_predict_false_when_exactly_on_the_limit
6,8,2009-08-01,2009,08,1,False,test_predict_false_when_exactly_on_the_limit
7,8,2009-09-01,2009,09,4500,False,test_predict_true_when_over_the_limit
7,8,2009-09-01,2009,09,0.01,True,test_predict_true_when_over_the_limit
5,8,2014-05-01,2014,05,7700,False,test_predict_false_when_under_the_limit
5,8,2014-05-01,2014,05,200,False,test_predict_false_when_under_the_limit
6,8,2014-06-01,2014,06,7999,False,test_predict_false_when_exactly_on_the_limit
6,8,2014-06-01,2014,06,1,False,test_predict_false_when_exactly_on_the_limit
7,8,2014-07-01,2014,07,7800,False,test_predict_true_when_over_the_limit
7,8,2014-07-01,2014,07,200.01,True,test_predict_true_when_over_the_limit

0 comments on commit d7ec25c

Please sign in to comment.