Binary classification example:

'''

Binary Classification.

'''

import numpy

import pandas

from microsoftml import rx_fast_linear, rx_predict

from revoscalepy.etl.RxDataStep import rx_data_step

from microsoftml.datasets.datasets import get_dataset

 

claims = get_dataset("claims")

 

import sklearn

if sklearn.__version__ < "0.18":

    from sklearn.cross_validation import train_test_split

else:

    from sklearn.model_selection import train_test_split

 

claimsdf = claims.as_df()

claimsdf["isCase"] = claimsdf.case == 1

data_train, data_test, y_train, y_test = train_test_split(claimsdf, claimsdf.isCase)

 

forest_model = rx_fast_linear(

    formula=" isCase ~ age + parity + education + spontaneous + induced ",

    data=data_train)

   

# RuntimeError: The type (RxTextData) for file is not supported.

score_ds = rx_predict(forest_model, data=data_test,

                     extra_vars_to_write=["isCase", "Score"])

                    

# Print the first five rows

print(rx_data_step(score_ds, number_rows_read=5))