Skip to content

Commit 05ab955

Browse files
j-groverJaskirat Grover
authored andcommitted
Update README.md
1 parent 561b33f commit 05ab955

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

README.md

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ pip uninstall autonormalize
3535

3636
### `auto_entityset`
3737
```shell
38-
auto_entityset(df, accuracy=0.98, index=None, name=None, time_index=None)
38+
auto_entityset(df, accuracy=0.98, index=None, name=None, time_index=None, variable_types=None)
3939
```
4040
Creates a normalized entityset from a dataframe.
4141

@@ -49,8 +49,10 @@ Creates a normalized entityset from a dataframe.
4949

5050
* `name` (str, optional) : the name of created EntitySet
5151

52-
* `time_index` (str, optional) : name of time column in the dataframe.
52+
* `time_index` (str, optional) : name of time column in the dataframe
5353

54+
* `variable_types` (dict[str -> Variable], optional) : Keys are of variable ids and values are variable types. Used to initialize an entity's store.
55+
5456
**Returns:**
5557

5658
* `entityset` (ft.EntitySet) : created entity set
@@ -85,7 +87,7 @@ Normalizes dataframe based on the dependencies given. Keys for the newly created
8587
### `make_entityset`
8688

8789
```shell
88-
make_entityset(df, dependencies, name=None, time_index=None)
90+
make_entityset(df, dependencies, name=None, time_index=None, variable_types=None)
8991
```
9092
Creates a normalized EntitySet from dataframe based on the dependencies given. Keys are chosen in the same fashion as for `normalize_dataframe`and a new index will be created if any key has more than a single attribute.
9193

0 commit comments

Comments
 (0)