Skip to content

Commit

Permalink
minor fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
pplonski committed Apr 9, 2019
1 parent e959f3c commit ce8acb4
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 4 deletions.
15 changes: 12 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ Thanks to Automated Machine Learning you don't need to worry about different mac

In the current version only binary classification is supported with optimization of LogLoss metric.

## Example
## Quick example

```python
import pandas as pd
Expand All @@ -28,12 +28,21 @@ automl = AutoML()
automl.fit(X, y)

predictions = automl.predict(X)


```

## The tuning algorithm

The tuning algorithm was created and developed by Piotr Płoński. It is heuristic algorithm created from combination of:

- **not-so-random** approach
- and **hill-climbing**

The approach is **not-so-random** because each algorithm has a defined set of hyper-parameters that usually works. At first step from not so random parameters an initial set of models is drawn. Then the hill climbing approach is used to pick best performing algorithms and tune them.

For each algorithm used in the AutoML the early stopping is applied.

The ensemble algorithm was implemented based on [Caruana paper](http://www.cs.cornell.edu/~alexn/papers/shotgun.icml04.revised.rev2.pdf).

## Installation

From PyPi repository:
Expand Down
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@

setup(
name='mljar-supervised',
version='0.1.0',
version='0.1.1',
description='Automated Machine Learning for Supervised tasks',
long_description=long_description,
long_description_content_type="text/markdown",
Expand Down

0 comments on commit ce8acb4

Please sign in to comment.