Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
hemangjoshi37a authored May 13, 2024
1 parent b4143a1 commit 03cd5ac
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -17,17 +17,18 @@ pip install TrendMaster
## Usage
Here's how to integrate TrendMaster into your Python projects:


```python
from trendmaster import TrendMaster
#Initialize the TrendMaster object
tm = TrendMaster(symbol_name_stk='SBIN')
#Load your data
test_symbol = 'SBIN'
tm = TrendMaster(symbol_name_stk=test_symbol)
#Load your data
data = tm.load_data(symbol=test_symbol)
#Train the model
tm.train(test_symbol, transformer_params={'epochs': 1})
#Perform inference
predictions = tm.inferencer.predict_future(val_data=data,future_steps=100,symbol='SBIN')
predictions = tm.inferencer.predict_future(val_data=data,future_steps=100,symbol=test_symbol)
print(predictions)
```

Expand Down

0 comments on commit 03cd5ac

Please sign in to comment.