pmlearn.linear_model.tests package

Submodules

pmlearn.linear_model.tests.test_base module

Testing for Linear regression

class pmlearn.linear_model.tests.test_base.TestLinearRegression[source]

Bases: object

setup_method()[source]
teardown_method()[source]
class pmlearn.linear_model.tests.test_base.TestLinearRegressionFit[source]

Bases: pmlearn.linear_model.tests.test_base.TestLinearRegression

test_advi_fit_returns_correct_model()[source]
test_nuts_fit_returns_correct_model()[source]
class pmlearn.linear_model.tests.test_base.TestLinearRegressionPredict[source]

Bases: pmlearn.linear_model.tests.test_base.TestLinearRegression

test_predict_raises_error_if_not_fit()[source]
test_predict_returns_mean_predictions_and_std()[source]
test_predict_returns_predictions()[source]
class pmlearn.linear_model.tests.test_base.TestLinearRegressionSaveandLoad[source]

Bases: pmlearn.linear_model.tests.test_base.TestLinearRegression

test_save_and_load_work_correctly()[source]
class pmlearn.linear_model.tests.test_base.TestLinearRegressionScore[source]

Bases: pmlearn.linear_model.tests.test_base.TestLinearRegression

test_score_matches_sklearn_performance()[source]

pmlearn.linear_model.tests.test_logistic module

Testing for Logistic regression

class pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegression[source]

Bases: object

setup_method()[source]
teardown_method()[source]
class pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegressionFit[source]

Bases: pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegression

test_advi_fit_returns_correct_model()[source]
class pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegressionPredict[source]

Bases: pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegression

test_predict_returns_predictions()[source]
class pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegressionPredictProba[source]

Bases: pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegression

test_predict_proba_raises_error_if_not_fit()[source]
test_predict_proba_returns_probabilities()[source]
test_predict_proba_returns_probabilities_and_std()[source]
class pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegressionSaveandLoad[source]

Bases: pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegression

test_save_and_load_work_correctly()[source]
class pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegressionScore[source]

Bases: pmlearn.linear_model.tests.test_logistic.TestHierarchicalLogisticRegression

test_score_scores()[source]

Module contents