def test_rf_algorithm(self): my_alg = RandomForestClassifier() response = my_alg.compute_prediction(self.input_data) self.assertEqual('OK', response['status']) self.assertTrue('label' in response) self.assertEqual('<=50K', response['label'])
def test_rf_algorithm(self): input_data = { "gender": "male", "race/ethnicity": "group A", "parental level of education": "bachelor's degree", "lunch": "standard", "math score": 99, "reading score": 99, "writing score": 99 } my_alg = RandomForestClassifier() response = my_alg.compute_prediction(input_data) self.assertNotEqual('OK', response['status']) self.assertFalse('label' in response)
def test_rf_algorithm(self): input_data = { "Gender": "Male", "Married": "Yes", "Dependents": 2, "Education": "Graduate", "Self_Employed": "Yes", "ApplicantIncome": 5849, "CoapplicantIncome": 6000, "LoanAmount": 120, "Loan_Amount_Term": 360, "Credit_History": 1, "Property_Area": "Urban", } my_alg = RandomForestClassifier() response = my_alg.compute_prediction(input_data) self.assertEqual("OK", response["status"]) self.assertTrue("label" in response) self.assertEqual("Approved", response["label"])
def test_rf_algorithm(self): input_data = { "Age": 37, "Workclass": "Private", "Education-Num": 9, "Marital Status": "Married-civ-spouse", "Occupation": "Craft-repair", "Relationship": "Husband", "Race": "White", "Sex": "Male", "Capital Gain": 0, "Capital Loss": 0, "Hours per week": 68, "Country": "United-States" } my_alg = RandomForestClassifier() response = my_alg.compute_prediction(input_data) self.assertEqual('OK', response['status']) self.assertTrue('label' in response) self.assertEqual(False, response['label'])
def test_rf_algorithm(self): input_data = {"age": 37, "workclass": "Private", "fnlwgt": 34146, "education": "HS-grad", "education-num": 9, "marital-status": "Married-civ-spouse", "occupation": "Craft-repair", "relationship": "Husband", "race": "White", "sex": "Male", "capital-gain": 0, "capital-loss": 0, "hours-per-week": 68, "native-country": "United-States"} my_alg = RandomForestClassifier() response = my_alg.compute_prediction(input_data) self.assertEqual("OK", response['status']) self.assertTrue("label" in response) self.assertEqual("<=50k", response["label"])
def test_rf_algorithm(self): input_data = { "age": 37, "workclass": "Private", "fnlwgt": 34146, "education": "HS-grad", "education-num": 9, "marital-status": "Married-civ-spouse", "occupation": "Craft-repair", "relationship": "Husband", "race": "White", "sex": "Male", "capital-gain": 0, "capital-loss": 0, "hours-per-week": 68, "native-country": "United-States" } my_alg = RandomForestClassifier() response = my_alg.compute_prediction(input_data) self.assertEqual('OK', response['status']) self.assertTrue('label' in response) self.assertEqual('<=50K', response['label']) def test_registry(self): registry = MLRegistry() self.assertEqual(len(registry.endpoints), 0) endpoint_name = "income_classifier" algorithm_object = RandomForestClassifier() algorithm_name = "random forest" algorithm_status = "production" algorithm_version = "0.0.1" algorithm_owner = "Piotr" algorithm_description = "Random Forest with simple pre- and post-processing" algorithm_code = inspect.getsource(RandomForestClassifier) # add to registry registry.add_algorithm(endpoint_name, algorithm_object, algorithm_name, algorithm_status, algorithm_version, algorithm_owner, algorithm_description, algorithm_code) # there should be one endpoint available self.assertEqual(len(registry.endpoints), 1)
def test_rf_algorithm(self): input_data = { "age": 48, "workclass": "Private", "fnlwgt": 171095, "education": "Assoc-acdm", "education-num": 12, "marital-status": "Divorced", "occupation": "Exec-managerial", "relationship": "Unmarried", "race": "White", "sex": "Female", "capital-gain": 0, "capital-loss": 0, "hours-per-week": 40, "native-country": "England" } my_alg = RandomForestClassifier() response = my_alg.compute_prediction(input_data) print(response) self.assertEqual('OK', response['status']) self.assertTrue('label' in response) self.assertEqual('<=50K', response['label'])